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
