소스 검색

add: cache

Ben 11 달 전
부모
커밋
5a8d8412cc
3개의 변경된 파일13개의 추가작업 그리고 2개의 파일을 삭제
  1. 11 1
      app.py
  2. 0 0
      data.json
  3. 2 1
      requirements.txt

+ 11 - 1
app.py

@@ -2,13 +2,21 @@ import logging
 
 import yt_dlp
 from flask import Flask, request
+from tinydb import TinyDB
+from tinydb.table import Document
 
 app = Flask(__name__)
 
+db = TinyDB('data.json')
+
 
 @app.route("/extract", methods=["GET", "POST"])
 def extract():
     url = request.json.get("url")
+    result = db.get(doc_id=url)
+    if result:
+        logging.info("find from data.json, so return")
+        return result
     logging.info(f"url: ${url}")
     with yt_dlp.YoutubeDL({"flat-playlist": True, "extract_flat": "flat-playlist"}) as ydl:
         info = ydl.extract_info(url, download=False)
@@ -40,7 +48,7 @@ def extract():
                     "abr": 0,
                     "container": item.get("container")
                 })
-        return {
+        result = {
             "code": 200,
             "msg": "",
             "data": {
@@ -63,6 +71,8 @@ def extract():
                 }
             }
         }
+        db.upsert(Document(value=result, doc_id=url))
+        return result
 
 
 if __name__ == '__main__':

파일 크기가 너무 크기때문에 변경 상태를 표시하지 않습니다.
+ 0 - 0
data.json


+ 2 - 1
requirements.txt

@@ -2,4 +2,5 @@ beautifulsoup4==4.12.2
 Flask==2.3.2
 Flask_Cors==3.0.10
 Requests==2.31.0
-yt_dlp==2024.4.9
+yt_dlp==2024.4.9
+tinydb==4.8.0

이 변경점에서 너무 많은 파일들이 변경되어 몇몇 파일들은 표시되지 않았습니다.