Supprimer bot/database.py
This commit is contained in:
parent
de1f1e9d55
commit
a9f2e56f63
@ -1,61 +0,0 @@
|
|||||||
import sqlite3
|
|
||||||
|
|
||||||
class UserLocation:
|
|
||||||
def __init__(self, db_name='users_location.db'):
|
|
||||||
self.db_name = db_name
|
|
||||||
self.create_table()
|
|
||||||
|
|
||||||
def create_table(self):
|
|
||||||
conn = sqlite3.connect(self.db_name)
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute('''
|
|
||||||
CREATE TABLE IF NOT EXISTS user_locations (
|
|
||||||
username TEXT NOT NULL,
|
|
||||||
user_id INTEGER PRIMARY KEY,
|
|
||||||
location TEXT NOT NULL
|
|
||||||
)
|
|
||||||
''')
|
|
||||||
conn.commit()
|
|
||||||
conn.close()
|
|
||||||
|
|
||||||
def set_user_location(self, username:str, user_id: int, location: str):
|
|
||||||
conn = sqlite3.connect(self.db_name)
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute('''
|
|
||||||
INSERT INTO user_locations (username, user_id, location) VALUES (?, ?, ?) ON CONFLICT(user_id) DO UPDATE SET location=excluded.location''', (username, user_id, location))
|
|
||||||
conn.commit()
|
|
||||||
conn.close()
|
|
||||||
|
|
||||||
def get_user_location(self, user_id: int) -> str:
|
|
||||||
conn = sqlite3.connect(self.db_name)
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute('SELECT location FROM user_locations WHERE user_id = ?', (user_id,))
|
|
||||||
row = cursor.fetchone()
|
|
||||||
conn.close()
|
|
||||||
return row[0] if row else None
|
|
||||||
|
|
||||||
class TaskCompleted:
|
|
||||||
def __init__(self, db_name='tasks_completed.db'):
|
|
||||||
self.db_name = db_name
|
|
||||||
self.create_table()
|
|
||||||
|
|
||||||
def create_table(self):
|
|
||||||
conn = sqlite3.connect(self.db_name)
|
|
||||||
cursor = conn.execute('''
|
|
||||||
CREATE TABLE IF NOT EXISTS task_completed (
|
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
||||||
user_id INTEGER NOT NULL,
|
|
||||||
username TEXT NOT NULL,
|
|
||||||
task TEXT NOT NULL,
|
|
||||||
time TEXT NOT NULL
|
|
||||||
)''')
|
|
||||||
conn.commit()
|
|
||||||
conn.close()
|
|
||||||
|
|
||||||
def set_info(self, user_id:int, username: str, task: str, time: str):
|
|
||||||
conn = sqlite3.connect(self.db_name)
|
|
||||||
cursor = conn.cursor()
|
|
||||||
cursor.execute('''
|
|
||||||
INSERT INTO task_completed (user_id, username, task, time) VALUES (?, ?, ?, ?)''', (user_id, username, task, time))
|
|
||||||
conn.commit()
|
|
||||||
conn.close()
|
|
||||||
Loading…
x
Reference in New Issue
Block a user