update initdb.py and learnapp.py to sqlalchemy 2.0

Miguel Barão
1 parent a79bce99
Showing 2 changed files with 13 additions and 12 deletions Show diff stats
aprendizations/initdb.py
aprendizations/learnapp.py
@@ -131,9 +131,9 @@ def main():
     # --- database stuff
     print(f'Database: {args.db}')
-    engine = create_engine(f'sqlite:///{args.db}', echo=False, future=True)
+    engine = create_engine(f'sqlite:///{args.db}')
     Base.metadata.create_all(engine)  # Creates schema if needed
-    session = Session(engine, future=True)
+    session = Session(engine)
     # --- make list of students to insert/update
     students = []
@@ -183,7 +183,7 @@ def main():
             count += 1
             print(f'  {sid}, {student.name}')
             passwd = (args.pw or sid).encode('utf-8')
-            student.password = bcrypt.hashpw(passwd, bcrypt.gensalt())
+            student.password = str(bcrypt.hashpw(passwd, bcrypt.gensalt()))
         session.commit()
         print(f'Total {count} password(s) updated.')
@@ -164,7 +164,7 @@ class LearnApp():
         await asyncio.sleep(random())
         query = select(Student).where(Student.id == uid)
-        with Session(self._engine, future=True) as session:
+        with Session(self._engine) as session:
             student = session.execute(query).scalar_one_or_none()
         if student is None:
             logger.info('User "%s" does not exist', uid)
@@ -186,7 +186,7 @@ class LearnApp():
             # get topics for this student and set its current state
             query = select(StudentTopic).where(StudentTopic.student_id == uid)
-            with Session(self._engine, future=True) as session:
+            with Session(self._engine) as session:
                 student_topics = session.execute(query).scalars().all()
             state = {t.topic_id: {
@@ -230,7 +230,7 @@ class LearnApp():
                                                bcrypt.gensalt())
         query = select(Student).where(Student.id == uid)
-        with Session(self._engine, future=True) as session:
+        with Session(self._engine) as session:
             session.execute(query).scalar_one().password = hashed_pw
             session.commit()
@@ -260,7 +260,7 @@ class LearnApp():
                         finishtime=str(question['finish_time']),
                         student_id=uid,
                         topic_id=topic_id)
-        with Session(self._engine, future=True) as session:
+        with Session(self._engine) as session:
             session.add(answer)
             session.commit()
@@ -285,7 +285,7 @@ class LearnApp():
             query = select(StudentTopic) \
                     .where(StudentTopic.student_id == uid) \
                     .where(StudentTopic.topic_id == tid)
-            with Session(self._engine, future=True) as session:
+            with Session(self._engine) as session:
                 student_topic = session.execute(query).scalar_one_or_none()
                 if student_topic is None:
                     # insert new studenttopic into database
@@ -343,7 +343,7 @@ class LearnApp():
         '''
         Fill table 'Topic' with topics from the graph, if new
         '''
-        with Session(self._engine, future=True) as session:
+        with Session(self._engine) as session:
             db_topics = session.execute(select(Topic.id)).scalars().all()
             new = [Topic(id=t) for t in topics if t not in db_topics]
             if new:
@@ -360,12 +360,13 @@ class LearnApp():
         if not exists(database):
             raise LearnException('Database does not exist')
-        self._engine = create_engine(f'sqlite:///{database}', future=True)
+        # echo=True enables logging of the SQL emitted by sqlalchemy
+        self._engine = create_engine(f'sqlite:///{database}', echo=False)
         try:
             query_students = select(func.count(Student.id))
             query_topics = select(func.count(Topic.id))
             query_answers = select(func.count(Answer.id))
-            with Session(self._engine, future=True) as session:
+            with Session(self._engine) as session:
                 count_students = session.execute(query_students).scalar()
                 count_topics = session.execute(query_topics).scalar()
                 count_answers = session.execute(query_answers).scalar()
@@ -584,7 +585,7 @@ class LearnApp():
                                     StudentTopic.topic_id,
                                     StudentTopic.level,
                                     StudentTopic.date)
-        with Session(self._engine, future=True) as session:
+        with Session(self._engine) as session:
             # all students in the database  FIXME only with answers of this course
             students = session.execute(query_students).all()
	@@ -131,9 +131,9 @@ def main():		@@ -131,9 +131,9 @@ def main():
131		131
132	# --- database stuff	132	# --- database stuff
133	print(f'Database: {args.db}')	133	print(f'Database: {args.db}')
134	- engine = create_engine(f'sqlite:///{args.db}', echo=False, future=True)	134	+ engine = create_engine(f'sqlite:///{args.db}')
135	Base.metadata.create_all(engine) # Creates schema if needed	135	Base.metadata.create_all(engine) # Creates schema if needed
136	- session = Session(engine, future=True)	136	+ session = Session(engine)
137		137
138	# --- make list of students to insert/update	138	# --- make list of students to insert/update
139	students = []	139	students = []
	@@ -183,7 +183,7 @@ def main():		@@ -183,7 +183,7 @@ def main():
183	count += 1	183	count += 1
184	print(f' {sid}, {student.name}')	184	print(f' {sid}, {student.name}')
185	passwd = (args.pw or sid).encode('utf-8')	185	passwd = (args.pw or sid).encode('utf-8')
186	- student.password = bcrypt.hashpw(passwd, bcrypt.gensalt())	186	+ student.password = str(bcrypt.hashpw(passwd, bcrypt.gensalt()))
187		187
188	session.commit()	188	session.commit()
189	print(f'Total {count} password(s) updated.')	189	print(f'Total {count} password(s) updated.')