feat: query add select_top_10_stock

src/fund_statistic.py

@@ -12,24 +12,14 @@ Copyright (c) 2020 Camel Lu
 import pymysql
 from pprint import pprint
 from db.connect import connect
+from sql_model.fund_query import FundQuery
 
 cursor = connect().cursor()
 
 if __name__ == '__main__':
-    page_start = 0
-    page_limit = 10000
-    stock_sql_join = ''
-    for index in range(10):
-        stock_sql_join = stock_sql_join + \
-            "t.top_stock_%s_code, t.top_stock_%s_name" % (
-                str(index), str(index)) + ","
-    # print(stock_sql_join[0:-1])
-    stock_sql_join = stock_sql_join[0:-1]
-    # print(stock_sql_join)
-    sql_query_season = "SELECT t.fund_code," + stock_sql_join + \
-        " FROM fund_morning_stock_info as t WHERE t.quarter_index = '2020-q4' AND t.stock_position_total > 20 LIMIT %s, %s ;"
-    cursor.execute(sql_query_season, [page_start, page_limit])    # 执行sql语句
-    results = cursor.fetchall()    # 获取查询的所有记录
+    each_query = FundQuery()
+    query_index = '2021-Q1'
+    results = each_query.select_top_10_stock(query_index)
     # pprint(results)
     code_dict = dict()
     for result in results:
@@ -49,6 +39,7 @@ if __name__ == '__main__':
             filer_dict[key] = value
             # print(key + ":" + str(value))
     list = sorted(filer_dict.items(), key=lambda x: x[1], reverse=True)
+    print(len(list))
     pprint(list)
     # pprint(dir(code_dict))
     # filer_dict = dict((name, getattr(code_dict, name))

src/sql_model/fund_query.py

@@ -85,8 +85,26 @@ class FundQuery:
         c_class_result = self.cursor.fetchone()
         return c_class_result
 
-    # 更新基金资产
+    # 更新基金资产 -- fund_morning_quarter
     def update_fund_total_asset(self, fund_code, total_asset):
         sql_update = "UPDATE fund_morning_quarter SET total_asset = %s WHERE fund_code = %s;"
         self.cursor.execute(sql_update, [total_asset, fund_code])
         self.connect_instance.commit()
+
+    def select_top_10_stock(self, query_index=None):
+        stock_sql_join = ''
+        for index in range(10):
+            stock_sql_join = stock_sql_join + \
+                "t.top_stock_%s_code, t.top_stock_%s_name" % (
+                    str(index), str(index)) + ","
+        # print(stock_sql_join[0:-1])
+        stock_sql_join = stock_sql_join[0:-1]
+        print(
+            "🚀 ~ file: fund_query.py ~ line 102 ~ stock_sql_join", stock_sql_join)
+        sql_query_season = "SELECT t.fund_code," + stock_sql_join + \
+            " FROM fund_morning_stock_info as t WHERE t.quarter_index = %s AND t.stock_position_total > 20;"  # 大于20%股票持仓基金
+        if query_index == None:
+            query_index = self.quarter_index
+        self.cursor.execute(sql_query_season, [query_index])    # 执行sql语句
+        results = self.cursor.fetchall()    # 获取查询的所有记录
+        return results