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!')
