mirror of
https://github.com/wahyd4/ai-projects.git
synced 2026-08-09 05:06:01 +10:00
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
import sqlite3
|
|
|
|
def init_db():
|
|
conn = sqlite3.connect('countries.db')
|
|
c = conn.cursor()
|
|
c.execute('''CREATE TABLE IF NOT EXISTS countries
|
|
(name TEXT PRIMARY KEY)''')
|
|
c.execute('''CREATE TABLE IF NOT EXISTS settings
|
|
(key TEXT PRIMARY KEY, value TEXT)''')
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
def add_country(country):
|
|
conn = sqlite3.connect('countries.db')
|
|
c = conn.cursor()
|
|
c.execute("INSERT OR IGNORE INTO countries (name) VALUES (?)", (country,))
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
def remove_country(country):
|
|
conn = sqlite3.connect('countries.db')
|
|
c = conn.cursor()
|
|
c.execute("DELETE FROM countries WHERE name = ?", (country,))
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
def get_all_countries():
|
|
conn = sqlite3.connect('countries.db')
|
|
c = conn.cursor()
|
|
c.execute("SELECT name FROM countries")
|
|
countries = [row[0] for row in c.fetchall()]
|
|
conn.close()
|
|
return countries
|
|
|
|
def save_highlight_color(color):
|
|
conn = sqlite3.connect('countries.db')
|
|
c = conn.cursor()
|
|
c.execute("INSERT OR REPLACE INTO settings (key, value) VALUES (?, ?)", ('highlight_color', color))
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
def get_highlight_color():
|
|
conn = sqlite3.connect('countries.db')
|
|
c = conn.cursor()
|
|
c.execute("SELECT value FROM settings WHERE key = 'highlight_color'")
|
|
result = c.fetchone()
|
|
conn.close()
|
|
return result[0] if result else None
|