deposti

service/api.py

@@ -6,6 +6,7 @@ from typing import Any
 from .llm import search_bot
 from .vector_store import vector_store
 from .logger import logger
+from util.time import TimeUtil
 
 
 class API():
@@ -69,5 +70,18 @@ class API():
         except Exception as e:
             print(e)
         return data
+    
+    def get_filecoin_deposit(self):
+        url="https://api.spacescope.io/v2/fvm/defi/all_protocol_flow?start_date={}&end_date={}".format(TimeUtil.get_date(offset=-2),TimeUtil.get_date(offset=-1))
+        headers = {
+        'authorization': f'Bearer {os.environ["SPACESCOPE_API_KEY"]}'
+        }
+        data = None
+        try:
+            response = requests.get(url, headers=headers)
+            data = json.loads(response.text)['data'][-1]['total_inflow_fil']
+        except Exception as e:
+            print(e)
+        return data
 
 api = API()

service/scheduler.py

@@ -10,18 +10,21 @@ class Scheduler:
 
     def __init__(self):
         self.scheduler = BackgroundScheduler()
-        self.jobs = [self.update_filecoin_price]
+        self.jobs = [self.update_filecoin_stat]
     
     def run(self):
         for job in self.jobs:
             self.scheduler.add_job(job, 'interval', minutes=5)
         self.scheduler.start()
 
-    def update_filecoin_price(self):
+    def update_filecoin_stat(self):
         price = api.get_filecoin_price()
-        if price is None:
-            return
-        text  = "The price of filecoin(FIL) is currently ${} USD.".format(price)
+        deposit = api.get_filecoin_deposit()
+        text  = """
+            The price of filecoin(FIL) is currently ${} USD.
+            The Total DeFi Net Deposits of filecoin(FIL) is {} FIL.
+        """.format(price,deposit)
         vector_store.upsert(text,"dataset_1")
+    
         
 

util/time.py

@@ -0,0 +1,22 @@
+import pytz
+from datetime import datetime,timedelta
+
+
+class TimeUtil:
+
+    @staticmethod
+    def get_date(offset=None):
+        beijing_tz = pytz.timezone('Asia/Shanghai')
+        time = datetime.now(beijing_tz)
+        if offset:
+            time = time +timedelta(days=offset)
+        format_date = time.strftime('%Y-%m-%d')
+        return format_date
+    
+    @staticmethod
+    def get_time():
+        beijing_tz = pytz.timezone('Asia/Shanghai')
+        beijing_time = datetime.now(beijing_tz)
+        format_time = beijing_time.strftime('%Y-%m-%d %H:%M:%S')
+        return format_time
+