add rahavard chart

Cuttlas90 · Jul 24, 2024 · 6342c88 · 6342c88
1 parent afbfe4a
commit 6342c88
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 4 deletions.
diff --git a/pages/helper/query.py b/pages/helper/query.py
@@ -24,16 +24,18 @@ def get_stock_data(self):
             """
         return string
 
-    def get_daily_social(self):
+    def get_daily_social(self, source):
         """return query to monthly sell value data"""
         string = f"""select
+                        index,
                         number,
                         date
                     from
                         public.social_data
                         INNER JOIN stocks ON public.social_data.stock_id = stocks.id
                     where
                         stocks.name = '{self.name}'
+                        and source = '{source}'
                     LIMIT
 						30
 

diff --git a/pages/social_page.py b/pages/social_page.py
@@ -32,13 +32,13 @@
 
     queries = Queries(name)
 
-    error, stock_data = vasahm_query(queries.get_daily_social())
+    error, stock_data = vasahm_query(queries.get_daily_social('sahamyab'))
     if error:
         st.error(stock_data, icon="🚨")
     else:
         if len(stock_data) > 0:
-            st.header("اقبال به سهم", divider='rainbow')
-            stock_data_history = pd.DataFrame(stock_data, columns=[
+            st.header("اقبال به سهم - سهامیاب", divider='rainbow')
+            stock_data_history = pd.DataFrame(stock_data, columns=["index",
             "number",
             "date"])
             # stock_data_history["date"] = stock_data_history[
@@ -50,3 +50,20 @@
                 alt.X('date:N',title="تاریخ")
             )
             st.altair_chart(chart, use_container_width=True)
+
+    error, stock_data = vasahm_query(queries.get_daily_social('rahavard'))
+    if error:
+        st.error(stock_data, icon="🚨")
+    else:
+        if len(stock_data) > 0:
+            st.header("اقبال به سهم - رهآورد", divider='rainbow')
+            stock_data_history = pd.DataFrame(stock_data, columns=["index",
+            "number",
+            "date"])
+
+            chart = alt.Chart(stock_data_history).mark_bar().encode(
+                alt.Color('index:N', title="سرفصلها"),
+                alt.Y('number:Q', title="تعداد کامنت"),
+                alt.X('date:N',title="تاریخ")
+            )
+            st.altair_chart(chart, use_container_width=True)