def link_effect_technology(): log.debug("Linking Effects with Technologies") if len(db_session.query(EffectTechnology).all()) <= 0: for tech in technologies: tech_o = db_session.query(Technology).filter(Technology.name == tech).first() for effect in blueprints[tech]: effect_o = Effect.query.filter(Effect.name == effect).first() effect_link = EffectTechnology(technology_id=tech_o.id, effect_id=effect_o.id) db_session.add(effect_link) db_session.commit()
def link_effect_event(): log.debug("Linking Effects with Events") if len(db_session.query(EffectEvent).all()) <= 0: for event in events: event_o = db_session.query(Event).filter(Event.name == event).first() for effect in blueprints[event]: effect_o = Effect.query.filter(Effect.name == effect).first() effect_link = EffectEvent(event_id=event_o.id, effect_id=effect_o.id) db_session.add(effect_link) db_session.commit()
def seed_producables(): log.debug("Seeding Producables") for pb in producables: if db_session.query(Producable).filter(Producable.name == pb).count() == 0: prd = Producable(name=pb, price=randint(4,59), time=randint(5,10)) db_session.add(prd) db_session.commit()
def seed_technology(): if len(db_session.query(Technology).all()) <= 0: log.debug("Adding Technology") for tex in technologies: tech = Technology(name=tex) db_session.add(tech) db_session.commit()
def seed_events(): if len(db_session.query(Event).all()) <= 0: log.debug("Adding Events") for evn in events: event = Event(name=evn) db_session.add(event) db_session.commit()
def seed_basegoods(): log.debug("Seeding Basegoods") for bg in basegoods: if db_session.query(BaseGood).filter(BaseGood.name == bg).count() == 0: basegood = BaseGood(name=bg, initprice=randint(1, 25), price=randint(4, 59)) db_session.add(basegood) db_session.commit()
def create_links(): log.debug("Creating Blueprints") if len(db_session.query(Blueprint).all()) <= 0: for prod in producables: # find producable by name -> get id # lookup in table to get the ammount and type # iterate over the basegoods # get quantity from table prod_o = db_session.query(Producable).filter(Producable.name == prod).first() for bg in blueprints[prod]: #bg_name, quant = bg.items()[0] # rather stupid python3 bg_name = list(bg.keys())[0] quant = bg[bg_name] basegood_o = BaseGood.query.filter(BaseGood.name == bg_name).first() for i in range(quant): blueprint = Blueprint(basegood_id=basegood_o.id, producable_id=prod_o.id) db_session.add(blueprint) db_session.commit()
def fill_inventory(): log.debug("Filling user inventories") for bg in basegoods: if db_session.query(Inventory).filter(BaseGood.name == bg).count() <= 0: basegood = BaseGood.query.filter(BaseGood.name == bg).first() all_users = User.query.all() for user in all_users: new_inv = Inventory(basegood=basegood, user_id=user.id, producable_id=None) db_session.add(new_inv) db_session.commit()
def seed_effects(): if len(db_session.query(Effect).all()) <= 0: log.debug("Adding Effects") #if db_session.query(Effect).filter(Effect.name == effects[0]).count() == 0: for efc in effects: # the return here is: # {'FastProduction': [{'description': 'that'}, {'name': 'that'}]} # dict of array of dicts # currently disabled for simplification effect = Effect(name=efc) db_session.add(effect) db_session.commit()
def seed_users(): log.debug("Seeding the Database") fake = Factory.create() log.debug("Seeding Users") for times in range(max_users): if db_session.query(User).count() <= max_users: usr = User(name=fake.name(), email=fake.email(), password=fake.state(), balance=randint(1000, 9000)) db_session.add(usr) db_session.commit()
def add_basegood_to_inv(user_id, bg_id): user = db_session.query(User).filter(User.id == user_id).first() bg = BaseGood.query.get(1) new_inv = Inventory(basegood=bg, user_id=user.id, producable_id=None) db_session.add(new_inv) db_session.commit()