Skip to content

Commit

Permalink
#100: Style changes
Browse files Browse the repository at this point in the history
  • Loading branch information
keiranjprice101 committed Oct 23, 2019
1 parent 6a52ea9 commit ae9d342
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions util/icat_db_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
from common.models import db_models
from common.session_manager import session_manager


parser = argparse.ArgumentParser()
parser.add_argument("--seed", "-s", dest="seed", help="Provide seed for random and faker", type=int, default=1)
parser.add_argument("--years", "-y", dest="years", help="Provide number of years to generate", type=int, default=20)
Expand Down Expand Up @@ -40,7 +39,7 @@ def get_date_time():
:return: the datetime
"""
return faker.date_time_between_dates(datetime_start=datetime.datetime(2000, 10, 4),
datetime_end=datetime.datetime(2019, 10, 5))
datetime_end=datetime.datetime(2019, 10, 5))


def get_start_date(i):
Expand Down Expand Up @@ -161,7 +160,7 @@ def generate_dataset_type(i):

class FacilityCycleGenerator(Generator):
tier = 1
amount = 4*YEARS # This gives 4 per year for 20 years
amount = 4 * YEARS # This gives 4 per year for 20 years

def generate(self):
self.pool_map(FacilityCycleGenerator.generate_facility_cycle)
Expand Down

0 comments on commit ae9d342

Please sign in to comment.