142 lines
4.4 KiB
Python
142 lines
4.4 KiB
Python
from datetime import datetime
|
|
|
|
|
|
class base_quest():
|
|
TABLE = None
|
|
|
|
def __init__(self, db, chat_id, qid, name=None, imp=None, diff=None,
|
|
state=None, date=None):
|
|
self.DB = db
|
|
self.CHAT_ID = chat_id
|
|
self.name = name
|
|
self.QID = qid
|
|
self.imp = imp
|
|
self.diff = diff
|
|
self.state = state
|
|
if date:
|
|
if isinstance(date, datetime):
|
|
self.date = date
|
|
else:
|
|
self.date = datetime.fromtimestamp(date)
|
|
else:
|
|
date = None
|
|
|
|
def add_to_db(self):
|
|
cursor = self.DB.cursor()
|
|
query = (f'INSERT INTO {self.TABLE}(chat_id, qid, name, importance'
|
|
', difficulty, date, state) values(?, ?, ?, ?, ?, ?, ?)')
|
|
cursor.execute(query, (self.CHAT_ID, self.QID, self.name, self.imp,
|
|
self.diff, self.date, self.state))
|
|
self.DB.commit()
|
|
|
|
def update_db(self):
|
|
cursor = self.DB.cursor()
|
|
query = (f'UPDATE {self.TABLE} SET name=?, importance=?, difficulty=?,'
|
|
' date=?, state=? WHERE chat_id=? AND qid=?')
|
|
cursor.execute(query, (self.name, self.imp, self.diff, self.date,
|
|
self.state, self.CHAT_ID, self.QID))
|
|
self.DB.commit()
|
|
|
|
def get_from_db(self):
|
|
cursor = self.DB.cursor()
|
|
query = (f'SELECT name, importance, difficulty, date, state FROM '
|
|
'{self.TABLE} where chat_id=? AND qid=?')
|
|
cursor.execute(query, self.CHAT_ID, self.QID)
|
|
output = cursor.fetchone()
|
|
self.name, self.imp, self.diff, self.date, self.state = output
|
|
|
|
|
|
class quest(base_quest):
|
|
TABLE = "quests"
|
|
|
|
|
|
class side_quest(base_quest):
|
|
TABLE = "side_quests"
|
|
|
|
|
|
def get_quest(db, chat_id, qid):
|
|
q = quest(db, chat_id, qid)
|
|
q.get_from_db()
|
|
return q
|
|
|
|
|
|
def add_quest(self, db, chat_id, qid, name=None, imp=None, diff=None,
|
|
state=None, date=None):
|
|
q = quest(self, db, chat_id, qid, name, imp, diff, state, date)
|
|
q.add_to_db()
|
|
return q
|
|
|
|
|
|
def get_side_quest(db, chat_id, qid):
|
|
q = side_quest(db, chat_id, qid)
|
|
q.get_from_db()
|
|
return q
|
|
|
|
|
|
def add_side_quest(self, db, chat_id, qid, name=None, imp=None, diff=None,
|
|
state=None, date=None):
|
|
q = side_quest(self, db, chat_id, qid, name, imp, diff, state, date)
|
|
q.add_to_db()
|
|
return q
|
|
|
|
|
|
class player():
|
|
def __init__(self, db, chat_id):
|
|
self.DB = db
|
|
self.CHAT_ID = chat_id
|
|
cursor = self.DB.cursor()
|
|
cursor.execute('SELECT * FROM state WHERE chat_id = ?')
|
|
row = cursor.fetchone()
|
|
if not row:
|
|
cursor.execute('INSERT INTO state(chat_id, state) VALUES(?,?)'
|
|
(chat_id, 'none'))
|
|
db.commit()
|
|
cursor.execute('SELECT * FROM points WHERE chat_id = ?')
|
|
row = cursor.fetchone()
|
|
if not row:
|
|
cursor.execute('INSERT INTO points(chat_id, points) VALUES(?,?)'
|
|
(chat_id, 0))
|
|
db.commit()
|
|
|
|
def get_state(self):
|
|
cursor = self.DB.cursor()
|
|
query = 'SELECT state FROM state WHERE chat_id=?'
|
|
cursor.execute(query, self.CHAT_ID)
|
|
output = cursor.fetchone()
|
|
return output[0]
|
|
|
|
def set_state(self, state):
|
|
cursor = self.DB.cursor()
|
|
query = 'UPDATE state SET state=? WHERE chat_id=?'
|
|
cursor.execute(query, (state, self.CHAT_ID))
|
|
self.DB.commit()
|
|
|
|
def get_points(self):
|
|
cursor = self.DB.cursor()
|
|
query = 'SELECT points FROM points WHERE chat_id=?'
|
|
cursor.execute(query, self.CHAT_ID)
|
|
output = cursor.fetchone()
|
|
return int(output[0])
|
|
|
|
def add_points(self, points):
|
|
new_points = self.get_points() + points
|
|
cursor = self.DB.cursor()
|
|
query = 'UPDATE points SET points=? WHERE chat_id=?'
|
|
cursor.execute(query, (new_points, self.CHAT_ID))
|
|
self.DB.commit()
|
|
|
|
def get_quests(self, state=1):
|
|
cursor = self.DB.cursor()
|
|
query = ('SELECT chat_id, qid, name, importance, difficulty, date, '
|
|
'state FROM quests WHERE chat_id = ?')
|
|
if state is not None:
|
|
query += ' AND state = ?'
|
|
cursor.execute(query, state)
|
|
else:
|
|
cursor.execute(query)
|
|
quests = []
|
|
for row in cursor:
|
|
q = quest(self.DB, *row)
|
|
quests.append(q)
|
|
return quests
|