def/MirrorTea
def/MirrorTea
Archived
1
1
Fork 0
Code Issues Pull requests Releases Activity

Test state: assertion error

Browse source
This commit is contained in:
localhost_frssoft 2023-01-19 16:13:10 +03:00
parent 1196424bbd
commit 9750ab5454
1 changed files with 6 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
mirrortea/repository/sqlite_repository.py
View file

@ -35,10 +35,8 @@ class SQLiteDatabaseRepository(AbstractDatabaseRepository):
self.conn = sqlite3.connect(self.path) self.conn = sqlite3.connect(self.path)
self.cursor = self.conn.cursor() self.cursor = self.conn.cursor()
def _namedtuple_factory(cursor, row, raw_tuples): def _namedtuple_factory(row, cursor, raw_tuples):
# TODO: fix that. Here causes errors. Or remove namedtyples and use normal index tuple. Also, row - is cursor fields = [column[0] for column in cursor.description]
# Good luck
fields = [column[0] for column in row.description]
cls = namedtuple("Row", fields) cls = namedtuple("Row", fields)
return cls._make(raw_tuples) return cls._make(raw_tuples)
@ -52,7 +50,7 @@ class SQLiteDatabaseRepository(AbstractDatabaseRepository):
def get_user_by_id(self, user_id: str) -> User: def get_user_by_id(self, user_id: str) -> User:
"""Get user by user id""" """Get user by user id"""
self.conn.row_factory = self._namedtuple_factory # Error here, not use named self.conn.row_factory = self._namedtuple_factory
self.cur_userdata = self.conn.execute( self.cur_userdata = self.conn.execute(
"SELECT user_id AS user_id, first_name AS first_name, last_name AS last_name, username AS username, avatar_hash AS avatar_hash FROM platforms_users WHERE user_id = (?)", "SELECT user_id AS user_id, first_name AS first_name, last_name AS last_name, username AS username, avatar_hash AS avatar_hash FROM platforms_users WHERE user_id = (?)",
@ -65,9 +63,9 @@ class SQLiteDatabaseRepository(AbstractDatabaseRepository):
return UserNotFoundError("User not found!") return UserNotFoundError("User not found!")
return User( return User(
id=user_id, user_id=self.userdata.user_id,
first_name=self.userdata.user_id, first_name=self.userdata.first_name,
last_name=self.userdata.first_name, last_name=self.userdata.last_name,
username=self.userdata.username, username=self.userdata.username,
avatar_hash=self.userdata.avatar_hash, avatar_hash=self.userdata.avatar_hash,
) )
@ -90,7 +88,6 @@ class SQLiteDatabaseRepository(AbstractDatabaseRepository):
return UserAlreadyExistError("User already exist!") return UserAlreadyExistError("User already exist!")
self.conn.commit() self.conn.commit()
return self.get_user_by_id(user_id=user.user_id) return self.get_user_by_id(user_id=user.user_id)
def update_user(self) -> None: def update_user(self) -> None: