def sample_users_articles(ssn: sa.orm.Session): """ Create a database with sample users and articles """ # Make some users ssn.add_all([ User(name='John', login='******', passwd='1', age=18, articles=[ Article(title='Jam'), Article(title='Jeep'), ]), User(name='Mark', login='******', passwd='2', age=20, articles=[ Article(title='Map'), Article(title='Mop'), ]), User(name='Nick', login='******', passwd='3', age=25, articles=[ Article(title='Nap'), Article(title='Nil'), ]), User(name='Kate', login='******', passwd='4', age=30, articles=[]), User(name='Cary', login='******', passwd='5', age=35, articles=[]), ]) ssn.commit()
def process_triggers(session: sqlalchemy.orm.Session, characters: List[Character], area: models.Area): """ Find triggers for an area, print them, and potentially resolve them. """ triggers = session.query(models.Trigger).filter( (~models.Trigger.resolved) & (models.Trigger.area_id == area.id) & (models.Trigger.party_lvl_min <= min(c.level for c in characters)) & (models.Trigger.party_size >= len(characters)) ).all() events = [t.event for t in triggers] for event in events: print('Event: {}'.format(event.name)) print(event.description) while 1: try: resolved_bool = prompt_bool('Resolved ?', True) except MalformedBoolException: continue break if resolved_bool: for t in event.triggers: t.resolved = True # TODO(jalex): Does this commit the triggers attached to the events? session.add_all(events) session.commit()
def init_characters(session: sqlalchemy.orm.Session) -> CharacterGroups: """ Initialize the party's characters. Returns: Character groups. """ characters = [] while 1: name = input('Character name: ') if not name: break chars = session.query( models.Character).filter(models.Character.name == name).all() if not chars: try: new_char = prompt_bool('New character?', False) except MalformedBoolException: continue if not new_char: continue player_name = input('Player name: ') char = models.Character(name=name, player_name=player_name) else: try: idx = prompt_choices( chars, lambda i, char: ' {} - {} ({})'.format( i, char.name, char.player_name)) except MalformedInputException: continue char = chars[idx] default_lvl = char.level or 1 try: char.level = int( input('Level: [{}] '.format(default_lvl)) or default_lvl) except ValueError: print('Supply integer level.') continue characters.append(char) character_groups = init_locations(session, characters) session.add_all(x for sl in character_groups for x in sl) session.commit() return character_groups
async def handle_push(connection: Connection, session: sqlalchemy.orm.Session) -> Optional[int]: msgs_got = 0 version: Optional[Version] = None async for msg in connection.socket: msgs_got += 1 msg_json = json.loads(msg) pushmsg = PushMessage(msg_json) # print(f"pushmsg: {msg}") if not pushmsg.operations: logger.warn("empty operations list in client PushMessage") for op in pushmsg.operations: logger.info(f"operation: {op}") # await connection.socket.send(f"answer is:{msg}") logger.info(f"message key={pushmsg.key}") latest_version_id = core.get_latest_version_id(session=session) logger.info( f"** version on server:{latest_version_id}, version in pushmsg:{pushmsg.latest_version_id}" ) if latest_version_id != pushmsg.latest_version_id: exc = f"version identifier isn't the latest one; " \ f"incoming: {pushmsg.latest_version_id}, on server:{latest_version_id}" if latest_version_id is None: logger.warn(exc) raise PushRejected(exc) if pushmsg.latest_version_id is None: logger.warn(exc) raise PullSuggested(exc) if pushmsg.latest_version_id < latest_version_id: logger.warn(exc) raise PullSuggested(exc) raise PushRejected(exc) if not pushmsg.islegit(session): raise PushRejected("message isn't properly signed") for listener in before_push: listener(session, pushmsg) # I) detect unique constraint conflicts and resolve them if possible unique_conflicts = find_unique_conflicts(pushmsg, session) conflicting_objects = set() for uc in unique_conflicts: obj = uc['object'] conflicting_objects.add(obj) for key, value in zip(uc['columns'], uc['new_values']): setattr(obj, key, value) for obj in conflicting_objects: make_transient(obj) # remove from session for model in set(type(obj) for obj in conflicting_objects): pk_name = get_pk(model) pks = [ getattr(obj, pk_name) for obj in conflicting_objects if type(obj) is model ] session.query(model).filter(getattr(model, pk_name).in_(pks)). \ delete(synchronize_session=False) # remove from the database session.add_all(conflicting_objects) # reinsert session.flush() # II) perform the operations operations = [ o for o in pushmsg.operations if o.tracked_model is not None ] post_operations: List[Tuple[Operation, SQLClass, Optional[SQLClass]]] = [] try: op: Operation for op in operations: (obj, old_obj) = await op.perform_async(pushmsg, session, pushmsg.node_id, connection.socket) if obj is not None: # if the op has been skipped, it wont be appended for post_operation handling post_operations.append((op, obj, old_obj)) resp = dict(type="info", op=dict( row_id=op.row_id, version=op.version, command=op.command, content_type_id=op.content_type_id, )) call_after_tracking_fn(session, op, obj) await connection.socket.send(json.dumps(resp)) except OperationError as e: logger.exception( "Couldn't perform operation in push from node %s.", pushmsg.node_id) raise PushRejected("at least one operation couldn't be performed", *e.args) # III) insert a new version if post_operations: # only if operations have been done -> create the new version version = Version(created=datetime.datetime.now(), node_id=pushmsg.node_id) session.add(version) # IV) insert the operations, discarding the 'order' column accomplished_operations = [ op for (op, obj, old_obj) in post_operations ] for op in sorted(accomplished_operations, key=attr('order')): new_op = Operation() for k in [k for k in properties_dict(op) if k != 'order']: setattr(new_op, k, getattr(op, k)) session.add(new_op) new_op.version = version session.flush() for op, obj, old_obj in post_operations: op.call_after_operation_fn(session, obj, old_obj) # from woodmaster.model.sql.model import WoodPile, Measurement # orphans = session.query(Measurement).filter(Measurement.woodpile_id == None).all() # print(f"orphans:{orphans}") for listener in after_push: listener(session, pushmsg) # return the new version id back to the client logger.info(f"version is: {version}") if version: await connection.socket.send( json.dumps( dict(type="result", new_version_id=version.version_id))) return {'new_version_id': version.version_id} else: await connection.socket.send( json.dumps(dict(type="result", new_version_id=None))) logger.info("sent nothing message") await connection.socket.close() logger.info("push ready")