Wall06 commited on
Commit
2474937
·
verified ·
1 Parent(s): 5915ea8

Delete modules/database.py

Browse files
Files changed (1) hide show
  1. modules/database.py +0 -68
modules/database.py DELETED
@@ -1,68 +0,0 @@
1
- import sqlite3
2
- import pandas as pd
3
- import hashlib
4
-
5
- DB_NAME = "menuvision.db"
6
-
7
- def init_db():
8
- conn = sqlite3.connect(DB_NAME)
9
- c = conn.cursor()
10
- # Table for Restaurants
11
- c.execute('''CREATE TABLE IF NOT EXISTS restaurants
12
- (id INTEGER PRIMARY KEY, name TEXT, location TEXT, owner TEXT)''')
13
-
14
- # Table for Users (Login/Signup)
15
- c.execute('''CREATE TABLE IF NOT EXISTS users
16
- (email TEXT PRIMARY KEY, password TEXT, role TEXT)''')
17
-
18
- conn.commit()
19
- conn.close()
20
-
21
- # --- USER FUNCTIONS ---
22
- def create_user(email, password, role):
23
- try:
24
- conn = sqlite3.connect(DB_NAME)
25
- c = conn.cursor()
26
- # Hash password for basic security
27
- hashed_pw = hashlib.sha256(password.encode()).hexdigest()
28
- c.execute("INSERT INTO users (email, password, role) VALUES (?, ?, ?)", (email, hashed_pw, role))
29
- conn.commit()
30
- conn.close()
31
- return f"✅ Account created for {email} as {role}!"
32
- except sqlite3.IntegrityError:
33
- return "❌ Error: User with this email already exists."
34
- except Exception as e:
35
- return f"Error: {e}"
36
-
37
- def verify_login(email, password):
38
- conn = sqlite3.connect(DB_NAME)
39
- c = conn.cursor()
40
- hashed_pw = hashlib.sha256(password.encode()).hexdigest()
41
-
42
- c.execute("SELECT role FROM users WHERE email=? AND password=?", (email, hashed_pw))
43
- result = c.fetchone()
44
- conn.close()
45
-
46
- if result:
47
- return f"✅ Login Successful! Welcome back, {result[0]}."
48
- else:
49
- return "❌ Invalid email or password."
50
-
51
- # --- RESTAURANT FUNCTIONS ---
52
- def add_restaurant(name, location, owner):
53
- try:
54
- conn = sqlite3.connect(DB_NAME)
55
- c = conn.cursor()
56
- c.execute("INSERT INTO restaurants (name, location, owner) VALUES (?, ?, ?)", (name, location, owner))
57
- conn.commit()
58
- conn.close()
59
- return f"Success: Restaurant '{name}' added!"
60
- except Exception as e:
61
- return f"Error: {e}"
62
-
63
- def search_restaurants(query):
64
- conn = sqlite3.connect(DB_NAME)
65
- sql = f"SELECT name, location, owner FROM restaurants WHERE name LIKE '%{query}%' OR location LIKE '%{query}%'"
66
- df = pd.read_sql_query(sql, conn)
67
- conn.close()
68
- return df