def create(db_session: Session, *, creator: DBUser) -> Game: game = Game(players=[creator], status=GameStatus.CREATED) game_manager.create_game(game.id) db_session.add(game) db_session.commit() db_session.refresh(game) return game
def create_by_user_id( self, user_id: UUID, obj_in: CandidateCreate, db: Session ) -> Candidate: db_candidate = self.model( name=obj_in.name, email=obj_in.email, linkedin_url=obj_in.linkedin_url, user_id=user_id, ) db.add(db_candidate) db.commit() db.refresh(db_candidate) return db_candidate
def dataset_index( *, db: Session = Depends(get_db), analysis_model_in: AnalysisModelIn ): dataset = crud.dataset.get(db_session=db, id=analysis_model_in.dataset_id) if not dataset: raise HTTPException(status_code=404, detail=f"dataset {analysis_model_in.dataset_id} not found") analysis_model = DBAnalysisModel(index="", name=analysis_model_in.name, dataset_id=analysis_model_in.dataset_id) db.add(analysis_model) db.commit() db.refresh(analysis_model) index_location = create_index(dataset, analysis_model_in.name, ANALYSIS_INDEX_DIR, analysis_model.id) analysis_model.index = index_location db.commit() return AnalysisModel(id=analysis_model.id, index=analysis_model.index, name=analysis_model.name, dataset_id=analysis_model.dataset_id)
def create_with_items(self, db: Session, transaction: TransactionCreate) -> Transaction: # Adding transaction db_trx = Transaction( type=TransactionTypes(transaction.type), description=transaction.description, amount=transaction.amount, user_id=transaction.user_id, ) db.add(db_trx) db.commit() db.refresh(db_trx) # Adding items for item in transaction.items: obj_in_data = jsonable_encoder(item) db_item = TransactionItem(**obj_in_data) db_item.transaction_id = db_trx.id db_item.product_id = db_item.product_id db.add(db_item) db.commit() db.refresh(db_trx) return db_trx