rc/aircox
13
1
Fork 1
Code Issues 15 Pull Requests Projects Releases Wiki Activity

add --next-month to avoid tricky bash things

Browse Source
This commit is contained in:
bkfox 2016-01-04 20:23:31 +01:00
parent f8156e3dc8
commit e75195e98c
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
programs/management/commands/diffusions_monitor.py
View File

@ -146,6 +146,11 @@ class Command (BaseCommand):
group.add_argument( group.add_argument(
'--month', type=int, default=now.month, '--month', type=int, default=now.month,
help='used by update, default is today\'s month') help='used by update, default is today\'s month')
group.add_argument(
'--next-month', action='store_true',
help='set the date to the next month of given date'
' (if next month from today'
)
group = parser.add_argument_group('mode') group = parser.add_argument_group('mode')
group.add_argument( group.add_argument(
@ -162,6 +167,13 @@ class Command (BaseCommand):
month = options.get('month'), month = options.get('month'),
day = 1) day = 1)
date = tz.make_aware(date) date = tz.make_aware(date)
if options.get('next_month'):
month = options.get('month')
date += tz.timedelta(days = 28)
if date.month == month:
date += tz.timedelta(days = 28)
date = date.replace(day = 1)
if options.get('update'): if options.get('update'):
Actions.update(date, mode = options.get('mode')) Actions.update(date, mode = options.get('mode'))