from flask import Flask, redirect, request, abort
import sqlite3
app = Flask(__name__)
DB = "links.db"
def get_db():
db = sqlite3.connect(DB)
db.row_factory = sqlite3.Row
return db
def init_db():
with get_db() as db:
db.execute("""
CREATE TABLE IF NOT EXISTS links (
id INTEGER PRIMARY KEY AUTOINCREMENT,
slug TEXT UNIQUE NOT NULL,
url TEXT NOT NULL,
label TEXT
)
""")
init_db()
HTML = """
Links
meus links
// clique para acessar
"""
@app.route("/")
def index():
return HTML
@app.route("/api/links")
def list_links():
with get_db() as db:
rows = db.execute("SELECT slug, url, label FROM links ORDER BY id DESC").fetchall()
return [dict(r) for r in rows]
@app.route("/api/links", methods=["POST"])
def add_link():
data = request.json
slug = data.get("slug", "").strip()
url = data.get("url", "").strip()
label = data.get("label", "").strip()
if not slug or not url:
return {"error": "slug e url obrigatórios"}, 400
try:
with get_db() as db:
db.execute("INSERT INTO links (slug, url, label) VALUES (?, ?, ?)", (slug, url, label))
return {"ok": True}
except sqlite3.IntegrityError:
return {"error": "slug já existe"}, 409
@app.route("/api/links/", methods=["DELETE"])
def delete_link(slug):
with get_db() as db:
db.execute("DELETE FROM links WHERE slug = ?", (slug,))
return {"ok": True}
@app.route("/")
def do_redirect(slug):
with get_db() as db:
row = db.execute("SELECT url FROM links WHERE slug = ?", (slug,)).fetchone()
if not row:
abort(404)
return redirect(row["url"])
if __name__ == "__main__":
app.run(host="0.0.0.0", port=5000)