def add_course(data): db = database.SessionLocal() course = Course(data['description']) db.add(course) db.commit() db.close() return None
def get_course(data): db = database.SessionLocal() course = Course.query.filter_by(description=data['description']).first() db.close() if course is None: return {'status': 'Course nao encontrado'} return {'course': str(course)}
def get_user(data): db = database.SessionLocal() user = User.query.filter_by(name=data['name']).first() db.close() if user is None: return {'status': 'User nao encontrado'} return {'user': str(user)}
def add_grade(data): db = database.SessionLocal() grade = Grade(data['idUser'], data['idCourse'], data['grade']) db.add(grade) db.commit() db.close() return None
def add_user(data): db = database.SessionLocal() user = User(data['name'], data['email']) db.add(user) db.commit() db.close() return None
async def mutate(root, info, id): try: flashcard = crud.delete_flashcard(db=database.SessionLocal(), id=id) return DeleteFlashcard(success=True, flashcard=flashcard) except: raise GraphQLError('The flashcard could not be deleted!')
def edit_user(data): db = database.SessionLocal() for i in data: if i == 'email': db.execute("UPDATE Users SET email = '" + data['email'] + "' WHERE name='" + data['name'] + "'") db.commit() db.close()
async def db_session_middleware(request: Request, call_next): response = Response("Internal server error", status_code=500) try: request.state.db = database.SessionLocal() response = await call_next(request) finally: request.state.db.close() return response
def get_db(): """ Helper to control database connection sessions. """ try: db_session = database.SessionLocal() yield db_session except SQLAlchemyError: print("Database session error.")
def get_grade(data): db = database.SessionLocal() grade = Grade.query.filter_by(idUser=data['idUser'], idCourse=data['idCourse']).first() db.close() if grade is None: return {'status': 'Grade nao encontrada'} return {'grade': str(grade)}
def delete_user(data): db = database.SessionLocal() user = User.query.filter_by(name=data['name']).first() current_db_sessions = db.object_session(user) current_db_sessions.delete(user) current_db_sessions.commit() current_db_sessions.close() return None
def edit_course(data): db = database.SessionLocal() for i in data: if i == 'description': db.execute("UPDATE Courses SET description = '" + data['description'] + "' WHERE id='" + data['id'] + "'") db.commit() db.close()
async def mutate(root, info, name): try: category_schema = schemas.CreateCategory(name=name) category = crud.create_category(db=database.SessionLocal(), category=category_schema) return CreateCategory(success=True, category=category) except: raise GraphQLError('The category could not be created!')
def delete_course(data): db = database.SessionLocal() course = Course.query.filter_by(description=data['description']).first() current_db_sessions = db.object_session(course) current_db_sessions.delete(course) current_db_sessions.commit() current_db_sessions.close() return None
def get_grades(): db = database.SessionLocal() grades = Grade.query.all() db.close() temp = [] for g in grades: jsonUser = {'grade': str(g)} temp.append(jsonUser) return temp
def delete_grade(data): db = database.SessionLocal() grade = Grade.query.filter_by(idUser=data['idUser'], idCourse=data['idCourse']).first() current_db_sessions = db.object_session(grade) current_db_sessions.delete(grade) current_db_sessions.commit() current_db_sessions.close() return None
def get_users(): db = database.SessionLocal() users = User.query.all() db.close() temp = [] for u in users: jsonUser = {'user': str(u)} temp.append(jsonUser) return temp
def get_courses(): db = database.SessionLocal() courses = Course.query.all() db.close() temp = [] for c in courses: jsonUser = {'course': str(c)} temp.append(jsonUser) return temp
async def mutate(root, info, question, answer, category_id): try: flashcard_schema = schemas.CreateFlashcard(question=question, answer=answer, category_id=category_id) flashcard = crud.create_flashcard(db=database.SessionLocal(), flashcard=flashcard_schema) return CreateFlashcard(success=True, flashcard=flashcard) except: raise GraphQLError('The flashcard could not be created!')
def get_report(): db = database.SessionLocal() grades = Grade.query.all() db.close() temp = [] for g in grades: jsonUser = { 'User': g.user.name, 'Course': g.course.description, 'Grade': g.grade } temp.append(jsonUser) return temp
def get_db(): db = database.SessionLocal() try: yield db finally: db.close()
async def mutate(root, info, id): try: category = crud.delete_category(db=database.SessionLocal(), id=id) return DeleteCategory(success=True, category=category) except: raise GraphQLError('The category could not be deleted!')
async def resolve_get_flashcard(root, info, id): try: flashcard = crud.get_flashcard(db=database.SessionLocal(), id=id) return flashcard except: raise GraphQLError('Flashcard not found!')
async def resolve_get_flashcards(root, info): try: flashcards = crud.get_flashcards(db=database.SessionLocal()) return flashcards except: raise GraphQLError('Flashcards not found!')
async def resolve_get_category(root, info, id): try: category = crud.get_category(db=database.SessionLocal(), id=id) return category except: raise GraphQLError('Category not found')
def get_aeroportos(user_data): db = database.SessionLocal()
async def resolve_get_categories(root, info): try: categories = crud.get_categories(db=database.SessionLocal()) return categories except: raise GraphQLError('Categories not found!')