darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-03 09:51:11 -05:00
Code Issues Packages Projects Releases Wiki Activity

unindent old fn

Browse Source
This commit is contained in:
Leo Hemsted
2017-08-31 14:28:44 +01:00
parent 3f34f8fa3e
commit 9fcd54c12b
1 changed files with 30 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
app/commands.py
View File

@@ -1,7 +1,7 @@
import uuid import uuid
from datetime import datetime from datetime import datetime, timedelta
from decimal import Decimal from decimal import Decimal
from flask_script import Command, Manager, Option from flask_script import Command, Option
from app import db from app import db
from app.dao.monthly_billing_dao import ( from app.dao.monthly_billing_dao import (
@@ -49,7 +49,7 @@ class PurgeFunctionalTestDataCommand(Command):
Option('-u', '-user-email-prefix', dest='user_email_prefix', help="Functional test user email prefix."), Option('-u', '-user-email-prefix', dest='user_email_prefix', help="Functional test user email prefix."),
) )
def run(self, service_name_prefix=None, user_email_prefix=None): def run(self, user_email_prefix=None):
if user_email_prefix: if user_email_prefix:
users = User.query.filter(User.email_address.like("{}%".format(user_email_prefix))).all() users = User.query.filter(User.email_address.like("{}%".format(user_email_prefix))).all()
for usr in users: for usr in users:
@@ -167,33 +167,33 @@ class CustomDbScript(Command):
class PopulateMonthlyBilling(Command): class PopulateMonthlyBilling(Command):
option_list = ( option_list = (
Option('-y', '-year', dest="year", help="Use for integer value for year, e.g. 2017"), Option('-y', '-year', dest="year", help="Use for integer value for year, e.g. 2017"),
)
def run(self, year):
service_ids = get_service_ids_that_need_billing_populated(
start_date=datetime(2016, 5, 1), end_date=datetime(2017, 8, 16)
) )
start, end = 1, 13
if year == '2016':
start = 4
def run(self, year): for service_id in service_ids:
service_ids = get_service_ids_that_need_billing_populated( print('Starting to populate data for service {}'.format(str(service_id)))
start_date=datetime(2016, 5, 1), end_date=datetime(2017, 8, 16) print('Starting populating monthly billing for {}'.format(year))
) for i in range(start, end):
start, end = 1, 13 print('Population for {}-{}'.format(i, year))
if year == '2016': self.populate(service_id, year, i)
start = 4
for service_id in service_ids: def populate(self, service_id, year, month):
print('Starting to populate data for service {}'.format(str(service_id))) create_or_update_monthly_billing(service_id, datetime(int(year), int(month), 1))
print('Starting populating monthly billing for {}'.format(year)) sms_res = get_monthly_billing_by_notification_type(
for i in range(start, end): service_id, datetime(int(year), int(month), 1), SMS_TYPE
print('Population for {}-{}'.format(i, year)) )
self.populate(service_id, year, i) email_res = get_monthly_billing_by_notification_type(
service_id, datetime(int(year), int(month), 1), EMAIL_TYPE
def populate(self, service_id, year, month): )
create_or_update_monthly_billing(service_id, datetime(int(year), int(month), 1)) print("Finished populating data for {} for service id {}".format(month, str(service_id)))
sms_res = get_monthly_billing_by_notification_type( print('SMS: {}'.format(sms_res.monthly_totals))
service_id, datetime(int(year), int(month), 1), SMS_TYPE print('Email: {}'.format(email_res.monthly_totals))
)
email_res = get_monthly_billing_by_notification_type(
service_id, datetime(int(year), int(month), 1), EMAIL_TYPE
)
print("Finished populating data for {} for service id {}".format(month, str(service_id)))
print('SMS: {}'.format(sms_res.monthly_totals))
print('Email: {}'.format(email_res.monthly_totals))