54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
import sqlite3
|
|
|
|
class DB:
|
|
def __init__(self, conn):
|
|
self.conn = conn
|
|
pass
|
|
|
|
def get_subforums(self):
|
|
return self._db().execute('select forum_id, name, description from subforums').fetchall()
|
|
|
|
def get_subforum(self, subforum):
|
|
return self._db().execute('select name, description from subforums where forum_id = ?', (subforum,)).fetchone()
|
|
|
|
def get_threads(self, subforum):
|
|
return self._db().execute('select thread_id, title from threads where forum_id = ?', (subforum,)).fetchall()
|
|
|
|
def get_thread(self, thread):
|
|
db = self._db()
|
|
title, text, author = db.execute('''
|
|
select title, text, name
|
|
from threads, users
|
|
where thread_id = ? and author_id = user_id
|
|
''',
|
|
(thread,)
|
|
).fetchone()
|
|
comments = db.execute('''
|
|
select comment_id, parent_id, name, text
|
|
from comments, users
|
|
where thread_id = ? and author_id = user_id
|
|
''',
|
|
(thread,)
|
|
).fetchall()
|
|
return title, text, author, comments
|
|
|
|
def get_comments(self, thread):
|
|
return self._db().execute('''
|
|
select text
|
|
from comments
|
|
where thread_id = ?
|
|
''',
|
|
(thread,)
|
|
).fetchall()
|
|
|
|
def get_comment_tree(self, comment):
|
|
db = self._db()
|
|
parent = db.execute('select text from comments where comment_id = ?', (comment,)).fetchall()
|
|
children = db.execute('select text from comments where parent_id = ?', (comment,)).fetchall()
|
|
print(parent, children)
|
|
return str(parent) + str(children)
|
|
return parent
|
|
|
|
def _db(self):
|
|
return sqlite3.connect(self.conn)
|