4
4
5
5
import dateutil .parser
6
6
from django .core .management .base import BaseCommand , CommandError
7
+ from django .utils import timezone
7
8
from holviapp .importer import HolviImporter
8
9
from holviapp .utils import list_invoices , list_orders
9
10
10
-
11
- def yesterday_proxy ():
12
- now_yesterday = datetime .datetime .now () - datetime .timedelta (days = 1 )
13
- start_yesterday = datetime .datetime .combine (now_yesterday .date (), datetime .datetime .min .time ())
14
- return start_yesterday .isoformat ()
11
+ from asylum .utils import datetime_proxy
15
12
16
13
17
14
class Command (BaseCommand ):
18
15
help = 'Import transaction data from Holvi API'
19
16
20
17
def add_arguments (self , parser ):
21
18
parser .add_argument ('--all' , action = 'store_true' , help = 'Import all Holvi transactions (WARNING: this may take forever)' )
22
- parser .add_argument ('since' , type = str , nargs = '?' , default = yesterday_proxy (), help = 'Import transactions updated since datetime, defaults to yesterday midnight' )
19
+ parser .add_argument ('since' , type = str , nargs = '?' , default = datetime_proxy (), help = 'Import transactions updated since datetime, defaults to yesterday midnight' )
23
20
24
21
def handle (self , * args , ** options ):
25
22
if (not options ['since' ]
@@ -28,7 +25,7 @@ def handle(self, *args, **options):
28
25
invoice_filters = {}
29
26
order_filters = {}
30
27
if not options .get ('all' , False ):
31
- since_parsed = dateutil .parser .parse (options ['since' ])
28
+ since_parsed = timezone . make_aware ( dateutil .parser .parse (options ['since' ]) )
32
29
if options ['verbosity' ] > 1 :
33
30
print ("Importing since %s" % since_parsed .isoformat ())
34
31
invoice_filters ['update_time_from' ] = since_parsed .isoformat ()
0 commit comments