45 lines
1.0 KiB
Python
45 lines
1.0 KiB
Python
import sqlite3
|
|
|
|
def connect():
|
|
return sqlite3.connect("cafe.db")
|
|
|
|
def setup_database():
|
|
db = connect()
|
|
cur = db.cursor()
|
|
|
|
# Table users
|
|
cur.execute("""
|
|
CREATE TABLE IF NOT EXISTS users (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
username TEXT,
|
|
password TEXT,
|
|
role TEXT
|
|
)
|
|
""")
|
|
|
|
# Table menu (buat nanti)
|
|
cur.execute("""
|
|
CREATE TABLE IF NOT EXISTS menu (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
nama TEXT,
|
|
kategori TEXT,
|
|
harga REAL,
|
|
stok INTEGER
|
|
)
|
|
""")
|
|
|
|
# Tambah user default
|
|
cur.execute("SELECT COUNT(*) FROM users")
|
|
if cur.fetchone()[0] == 0:
|
|
users = [
|
|
("admin", "admin", "admin"),
|
|
("kasir", "kasir", "kasir"),
|
|
("waiter", "waiter", "waiter"),
|
|
("pembeli", "pembeli", "pembeli"),
|
|
("pemilik", "pemilik", "pemilik cafe")
|
|
]
|
|
cur.executemany("INSERT INTO users(username,password,role) VALUES (?,?,?)", users)
|
|
db.commit()
|
|
|
|
db.close()
|