Your Name 4 年之前
父節點
當前提交
11888571c2
共有 1 個文件被更改,包括 15 次插入1 次删除
  1. 15 1
      hhh/main.py

+ 15 - 1
hhh/main.py

@@ -2,13 +2,27 @@ from fastapi import FastAPI
 import sys
 import os
 import dataset
+import requests
 db = dataset.connect('sqlite:///tmp/hhh.db')
 
 app = FastAPI()
 
+
+
+def check_db(domain_name):
+    global db
+    table=db['q_cache']
+    table.upsert({'query':domain_name},keys=['query'])
+    db.commit()
+#    db.query('select')
+
 @app.get("/domain/{domain_name}")
 async def test_domain(domain_name):
-    return {"item_id": domain_name}
+    check_db(domain_name)
+    r = requests.get('https://data.similarweb.com/api/v1/data?domain='+domain_name)
+    js=r.json()
+    #https://data.similarweb.com/api/v1/data?domain=hhh.com.tw
+    return {"item_id": js}
 
 @app.get("/")
 async def root():