sim1222/fedi-block-api
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

also use hash to find blocks

Browse Source
This commit is contained in:
Enju Aihara 2022-04-22 14:58:23 +02:00
parent 1461b53de3
commit 8e4ba7958e
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api.py
View File

@ -1,9 +1,13 @@
from fastapi import FastAPI from fastapi import FastAPI
import sqlite3 import sqlite3
from hashlib import sha256
base_url = "" base_url = ""
app = FastAPI(docs_url=base_url+"/docs", redoc_url=base_url+"/redoc") app = FastAPI(docs_url=base_url+"/docs", redoc_url=base_url+"/redoc")
def get_hash(domain: str) -> str:
return sha256(domain.encode("utf-8")).hexdigest()
@app.get(base_url+"/info") @app.get(base_url+"/info")
def info(): def info():
conn = sqlite3.connect("blocks.db") conn = sqlite3.connect("blocks.db")
@ -23,7 +27,7 @@ def blocked(domain: str):
conn = sqlite3.connect("blocks.db") conn = sqlite3.connect("blocks.db")
c = conn.cursor() c = conn.cursor()
wildchar = "*." + ".".join(domain.split(".")[-domain.count("."):]) wildchar = "*." + ".".join(domain.split(".")[-domain.count("."):])
c.execute("select blocker, block_level, reason from blocks where blocked = ? or blocked = ?", (domain, wildchar)) c.execute("select blocker, block_level, reason from blocks where blocked = ? or blocked = ? or blocked = ?", (domain, wildchar, get_hash(domain)))
blocks = c.fetchall() blocks = c.fetchall()
conn.close() conn.close()