main.py 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. import rpyc
  2. import time
  3. from fastapi import FastAPI
  4. import sys
  5. import os
  6. import dataset
  7. import datetime
  8. import json
  9. import ast
  10. from fastapi.middleware.cors import CORSMiddleware
  11. db = dataset.connect('sqlite:///similar.db')
  12. app = FastAPI()
  13. origins = [
  14. "http://www.googo.org",
  15. "http://www.googo.org:8080",
  16. "http://googo.org:8080",
  17. "http://googo.org",
  18. "http://139.162.121.30"
  19. ]
  20. app.add_middleware(
  21. CORSMiddleware,
  22. allow_origins=origins,
  23. allow_credentials=True,
  24. allow_methods=["*"],
  25. allow_headers=["*"],
  26. )
  27. @app.get("/domain/{domain_name}")
  28. async def read_item(domain_name):
  29. cursor=db.query('select domain_name,json,dt from similar where domain_name="'+domain_name+'"')
  30. for c in cursor:
  31. print(c['json'])
  32. print(c['dt'])
  33. # jsdict=ast.literal_eval(c['json'])
  34. jsdict=json.loads(c['json'])
  35. return jsdict
  36. print(domain_name)
  37. # conn = rpyc.connect("localhost",12345)
  38. conn = rpyc.connect("192.168.192.119",12345)
  39. #domain_name
  40. jsdata=conn.root.get_url('https://data.similarweb.com/api/v1/data?domain='+domain_name)
  41. print(jsdata)
  42. # jsdict=str(jsdata)[1:-1]
  43. # jsdict=json.loads(str(jsdata))
  44. jsdict=ast.literal_eval(str(jsdata))
  45. # jsdict=str(jsdata)
  46. table=db['similar']
  47. jstr=json.dumps(jsdict)
  48. table.insert({'domain_name':domain_name,'json':jstr,'dt':datetime.datetime.now()})
  49. db.commit()
  50. return jsdata
  51. # print(jsdata)
  52. # return {"item_id": domain_name}