Add player class, status functions to it.

This commit is contained in:
Ceda EI 2018-10-28 16:04:52 +05:30
parent a3f50b5334
commit 84c3e94c0f
1 changed files with 32 additions and 0 deletions

View File

@ -78,3 +78,35 @@ def add_side_quest(self, db, chat_id, qid, name=None, imp=None, diff=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()