history
This commit is contained in:
parent
6c2da5e767
commit
465bcaf125
94
src/history.py
Normal file
94
src/history.py
Normal file
@ -0,0 +1,94 @@
|
||||
# coding=utf-8
|
||||
import sqlite3
|
||||
from settings import Settings
|
||||
import os
|
||||
|
||||
|
||||
class History(object):
|
||||
def __init__(self, name):
|
||||
self._name = name
|
||||
os.chdir(Settings.get_default_path())
|
||||
db = sqlite3.connect(name + '.hstr')
|
||||
cursor = db.cursor()
|
||||
cursor.execute('CREATE TABLE IF NOT EXISTS friends('
|
||||
' tox_id TEXT PRIMARY KEY'
|
||||
')')
|
||||
db.close()
|
||||
|
||||
def add_friend_to_db(self, tox_id):
|
||||
try:
|
||||
os.chdir(Settings.get_default_path())
|
||||
db = sqlite3.connect(self._name + '.hstr')
|
||||
cursor = db.cursor()
|
||||
cursor.execute('INSERT INTO friends VALUES (?);', (tox_id, ))
|
||||
cursor.execute('CREATE TABLE id' + tox_id + '('
|
||||
' id INTEGER PRIMARY KEY,'
|
||||
' message TEXT,'
|
||||
' owner INTEGER,'
|
||||
' unix_time INTEGER'
|
||||
' message_type INTEGER'
|
||||
')')
|
||||
db.commit()
|
||||
except Exception as e:
|
||||
db.rollback()
|
||||
raise e
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
def delete_friend_from_db(self, tox_id):
|
||||
try:
|
||||
os.chdir(Settings.get_default_path())
|
||||
db = sqlite3.connect(self._name + '.hstr')
|
||||
cursor = db.cursor()
|
||||
cursor.execute('DELETE FROM friends WHERE tox_id=?;', (tox_id, ))
|
||||
cursor.execute('DROP TABLE id' + tox_id + ';')
|
||||
db.commit()
|
||||
except Exception as e:
|
||||
db.rollback()
|
||||
raise e
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
def save_messages_to_db(self, tox_id, messages_iter):
|
||||
try:
|
||||
os.chdir(Settings.get_default_path())
|
||||
db = sqlite3.connect(self._name + '.hstr')
|
||||
cursor = db.cursor()
|
||||
cursor.executemany('INSERT INTO id' + tox_id + '(message, owner, unix_time) '
|
||||
'VALUES (?, ?, ?);', messages_iter)
|
||||
db.commit()
|
||||
except Exception as e:
|
||||
db.rollback()
|
||||
raise e
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
def messages_getter(self, tox_id):
|
||||
return MessageGetter(self._name, tox_id)
|
||||
|
||||
|
||||
class MessageGetter(object):
|
||||
def __init__(self, name, tox_id):
|
||||
os.chdir(Settings.get_default_path())
|
||||
self._db = sqlite3.connect(name + '.hstr')
|
||||
self._cursor = self._db.cursor()
|
||||
self._cursor.execute('SELECT message, owner, unix_time FROM id' + tox_id +
|
||||
' ORDER BY unix_time DESC;')
|
||||
|
||||
def get_one(self):
|
||||
return self._cursor.fetchone()
|
||||
|
||||
def get_all(self):
|
||||
return self._cursor.fetchall()
|
||||
|
||||
def get(self, count):
|
||||
return self._cursor.fetchmany(count)
|
||||
|
||||
def __del__(self):
|
||||
self._db.close()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
h = History('test')
|
||||
getter = h.messages_getter('42')
|
||||
print getter.get_all()
|
Loading…
Reference in New Issue
Block a user