30 lines
789 B
Python
30 lines
789 B
Python
|
from erminig.models.db import ErminigDB
|
||
|
|
|
||
|
|
|
||
|
|
def upsert_upstream(db: ErminigDB, name, type_, url, pattern, file):
|
||
|
|
db.cursor.execute(
|
||
|
|
"""
|
||
|
|
INSERT INTO upstreams (name, type, url, pattern, file, created_at)
|
||
|
|
VALUES (?, ?, ?, ?, ?, CURRENT_TIMESTAMP)
|
||
|
|
ON CONFLICT(name) DO UPDATE SET
|
||
|
|
type=excluded.type,
|
||
|
|
url=excluded.url,
|
||
|
|
pattern=excluded.pattern,
|
||
|
|
file=excluded.file
|
||
|
|
RETURNING id
|
||
|
|
""",
|
||
|
|
(name, type_, url, pattern, file),
|
||
|
|
)
|
||
|
|
row = db.cursor.fetchone()
|
||
|
|
return row[0] if row else None
|
||
|
|
|
||
|
|
|
||
|
|
def get_all_upstreams(db: ErminigDB):
|
||
|
|
db.cursor.execute(
|
||
|
|
"""
|
||
|
|
SELECT id, name, type, url, pattern, file FROM upstreams
|
||
|
|
"""
|
||
|
|
)
|
||
|
|
rows = db.cursor.fetchall()
|
||
|
|
return rows
|