73 lines
1.9 KiB
Python
73 lines
1.9 KiB
Python
from datetime import datetime
|
|
import hashlib
|
|
import sqlite3
|
|
|
|
|
|
def con3():
|
|
conn = sqlite3.connect('db/db.sqlite')
|
|
return conn
|
|
|
|
|
|
def create_user(name: str, email: str, password: str):
|
|
password = hashlib.sha256(password.encode()).hexdigest()
|
|
now = datetime.now().isoformat()
|
|
query = (f"INSERT INTO users (name, email, password, created_at, updated_at) VALUES ('{name}', '{email}', "
|
|
f"'{password}', '{now}', '{now}');")
|
|
conn = con3()
|
|
cursor = conn.cursor()
|
|
cursor.execute(query)
|
|
conn.commit()
|
|
conn.close()
|
|
return cursor.lastrowid
|
|
|
|
|
|
def get_user(id: int):
|
|
query = f"SELECT * FROM users WHERE id = {id};"
|
|
conn = con3()
|
|
cursor = conn.cursor()
|
|
cursor.execute(query)
|
|
user = cursor.fetchone()
|
|
conn.close()
|
|
return user
|
|
|
|
|
|
def get_user_by_email(email: str):
|
|
query = f"SELECT * FROM users WHERE email = '{email}';"
|
|
conn = con3()
|
|
cursor = conn.cursor()
|
|
cursor.execute(query)
|
|
user = cursor.fetchone()
|
|
conn.close()
|
|
return user
|
|
|
|
|
|
def create_habit(name: str, user_id: int, times: int, unit: int, note: str | None=None):
|
|
now = datetime.now().isoformat()
|
|
query = (f"INSERT INTO habits (user_id, name, note, times, unit, created_at, updated_at) VALUES ('{user_id}', "
|
|
f"'{name}', '{note}', '{times}', '{unit}', '{now}', '{now}');")
|
|
conn = con3()
|
|
cursor = conn.cursor()
|
|
cursor.execute(query)
|
|
conn.commit()
|
|
conn.close()
|
|
return cursor.lastrowid
|
|
|
|
|
|
def get_habit(id: int):
|
|
query = f"SELECT * FROM habits WHERE id = {id};"
|
|
conn = con3()
|
|
cursor = conn.cursor()
|
|
cursor.execute(query)
|
|
habits = cursor.fetchone()
|
|
conn.close()
|
|
return habits
|
|
|
|
|
|
def get_habits(user_id: int):
|
|
query = f"SELECT * FROM habits WHERE user_id = {user_id};"
|
|
conn = con3()
|
|
cursor = conn.cursor()
|
|
cursor.execute(query)
|
|
habits = cursor.fetchone()
|
|
conn.close()
|
|
return habits |