LBRYCommunity/comment-server
1
0
Fork 0
Code Issues 18 Pull requests 1 Projects Releases Packages Wiki Activity

Adds backup code + fixes obtain comment by reply code

Browse source
This commit is contained in:
Oleg Silkin 2019-05-20 01:17:32 -04:00
parent 1e38fd8ca3
commit 85df5c86cf
1 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
server/database.py
View file

@ -35,6 +35,10 @@ class DatabaseConnection:
with open(filepath, 'r') as ddl_file: with open(filepath, 'r') as ddl_file:
self.connection.executescript(ddl_file.read()) self.connection.executescript(ddl_file.read())
def create_backup(self):
with sqlite3.connect(backup_dir) as back:
self.connection.backup(back)
def get_claim_comments(self, claim_id: str, def get_claim_comments(self, claim_id: str,
parent_id: str = None, parent_id: str = None,
page: int = 1, page: int = 1,
@ -142,15 +146,12 @@ class DatabaseConnection:
def get_comments_by_id(self, comment_ids: list) -> typing.Union[list, None]: def get_comments_by_id(self, comment_ids: list) -> typing.Union[list, None]:
""" Returns a list containing the comment data associated with each ID within the list""" """ Returns a list containing the comment data associated with each ID within the list"""
if len(comment_ids) == 0: # format the input, under the assumption that the
return None
placeholders = ', '.join('?' for _ in comment_ids) placeholders = ', '.join('?' for _ in comment_ids)
curs = self.connection.execute( return [dict(row) for row in self.connection.execute(
f'SELECT * FROM COMMENTS_ON_CLAIMS WHERE comment_id IN ({placeholders})', f'SELECT * FROM COMMENTS_ON_CLAIMS WHERE comment_id IN ({placeholders})',
tuple(comment_ids) tuple(comment_ids)
)]
)
return [dict(row) for row in curs.fetchall()]
if __name__ == '__main__': if __name__ == '__main__':