fig getQifName -> takes datetime.datetime, not str

lib/browser.py

@@ -12,6 +12,7 @@
 #
 logged_in_urls = ['https://ib.nab.com.au/nabib/acctInfo_acctBal.ctl',
                   'https://ib.nab.com.au/nabib/loginProcess.ctl']
+TRANSACTIONS_PER_PAGE = 200
 
 
 def get_accounts(text):
@@ -234,7 +235,7 @@ def get_servers_today_date(b):
 
 
 def query_server_transactions(b, start_date):
-    TRANSACTIONS_PER_PAGE = 200
+
     end_date = get_servers_today_date(b)
     b.select_form(name='transactionHistoryForm')
     b.form['periodModeSelect'] = ['Custom']
@@ -258,7 +259,6 @@ def query_server_transactions(b, start_date):
 
 
 def get_all_transactions(b, account, start_date):
-    TRANSACTIONS_PER_PAGE = 200
     b = query_server_transactions(b, start_date)
     if not b:
         return None

lib/qif.py

@@ -2,18 +2,16 @@
 
 import re
 import os
-import time
+import datetime
 
 
 def get_qif_name(start_date, end_date):
-
-    date_format = '%Y.%m.%d'
-    start = time.strptime(start_date, "%d %b %y")
-    end = time.strptime(end_date, "%d %b %y")
-
-    return '%d.%d.%d - %d.%d.%d.qif' % (
-        start[0], start[1], start[2],
-        end[0], end[1], end[2],
+    """
+    Takes 2 datetime.datetime objects and returns a string representing the QIF file name
+    """
+    return '20%s - 20%s.qif' % (
+        datetime.datetime.strftime(start_date, "%y.%m.%d"),
+        datetime.datetime.strftime(end_date, "%y.%m.%d"),
     )
 
 

test.py

@@ -3,6 +3,7 @@
 import unittest
 
 from lib import qif
+import datetime
 
 
 class TestQIF(unittest.TestCase):
@@ -12,8 +13,11 @@ def setUp(self):
 
     def test_get_qif_name(self):
 
-        name = qif.get_qif_name('1 Jan 10', '13 Apr 12')
-        self.assertEqual(name, '2010.1.1 - 2012.4.13.qif')
+        name = qif.get_qif_name(
+            datetime.datetime(2010, 1, 1, 00, 00, 00),
+            datetime.datetime(2012, 4, 13, 00, 00, 00)
+        )
+        self.assertEqual(name, '2010.01.01 - 2012.04.13.qif')
 
     def test_is_file_present(self):