chinadata 0.3.6__py3-none-any.whl → 0.3.8__py3-none-any.whl

chinadata/ca_data.py

@@ -753,19 +753,74 @@ class pro_api:
         return self.query(token=self.token, api_name=api_name, **kwargs)
     def ft_limit(self, api_name='ft_limit', **kwargs):
         return self.query(token=self.token, api_name=api_name, **kwargs)
+
+    def request_with_retry(self,url, max_retries,):
+        for attempt in range(max_retries):
+            try:
+                response = requests.get(url=url, timeout=100, )
+                response.raise_for_status()
+                return response
+            except Exception as e:
+                # print(f"尝试 {attempt + 1} 失败: {e}")
+                if attempt < max_retries - 1:
+                    wait_time = 1 * (2 ** attempt)  # 指数退避
+                    # print(f"等待 {wait_time} 秒后重试...")
+                    time.sleep(wait_time)
+        # print(f"请求失败，已重试 {max_retries} 次")
+        return None
+
+    def rt_min(self,ts_code,freq,max_retries=3):
+        url = f"http://39.105.209.102:9002/csi/stock/min/rt_min/{ts_code}/{freq}/{self.token}"
+        r=self.request_with_retry(url, max_retries=max_retries)
+        df=pd.DataFrame(r.json())
+        df['time'] = pd.to_datetime(df['time'])
+        if freq == '60min':
+            periods = {
+                '09:15-10:30': ('09:15:00', '10:30:00'),
+                '10:31-11:30': ('10:31:00', '11:30:00'),
+                '13:01-14:00': ('13:01:00', '14:00:00'),
+                '14:01-15:00': ('14:01:00', '15:00:00')
+            }
+            result = []
+            for name, (start, end) in periods.items():
+                mask = (df['time'].dt.strftime('%H:%M:%S') >= start) & \
+                       (df['time'].dt.strftime('%H:%M:%S') <= end)
+                data = df[mask]
+                if len(data) > 0:
+                    result.append({
+                        'ts_code': df.iloc[0]['ts_code'],
+                        'time': end,
+                        'open': data.iloc[0]['open'],
+                        'high': data['high'].max(),
+                        'low': data['low'].min(),
+                        'close': data.iloc[-1]['close'],
+                        'vol': data['vol'].sum(),
+                        'amount': data['amount'].sum()
+                    })
+            df = pd.DataFrame(result)
+
+        else:
+            # 设置为索引
+            df.set_index('time', inplace=True)
+            df = df.resample(freq, label='right', closed='right').agg({
+                'ts_code': 'first',
+                'open': 'first',
+                'high': 'max',
+                'low': 'min',
+                'close': 'last',
+                'vol': 'sum',
+                'amount': 'sum',
+            }).dropna().reset_index()[[ 'ts_code','time','open','high','low','close','vol','amount']]
+
+        return df
     def irm_qa_sh(self, api_name='irm_qa_sh', **kwargs):
         return self.query(token=self.token, api_name=api_name, **kwargs)
-
     def irm_qa_sz(self, api_name='irm_qa_sz', **kwargs):
         return self.query(token=self.token, api_name=api_name, **kwargs)
-
     def stk_nineturn(self, api_name='stk_nineturn', **kwargs):
         return self.query(token=self.token, api_name=api_name, **kwargs)
-
-
     def stk_week_month_adj(self, api_name='stk_week_month_adj', **kwargs):
         return self.query(token=self.token, api_name=api_name, **kwargs)
-
     def limit_list_ths(self, api_name='limit_list_ths', **kwargs):
         return self.query(token=self.token, api_name=api_name, **kwargs)
 

{chinadata-0.3.6.dist-info → chinadata-0.3.8.dist-info}/METADATA

@@ -1,9 +1,8 @@
 Metadata-Version: 2.1
 Name: chinadata
-Version: 0.3.6
+Version: 0.3.8
 Author: xc
 Author-email: utyu@gmail.com
 Platform: any
 Requires-Dist: requests
 Requires-Dist: pandas
-

chinadata-0.3.8.dist-info/RECORD

@@ -0,0 +1,7 @@
+chinadata/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+chinadata/ca_data.py,sha256=sFzL91lw0_AF8wnFozU_Iu7KPvCzkepK2JlPdBPtOqg,40118
+chinadata/example.py,sha256=eKj2Hl5KVq9EJGWVY2ZoRJOqZ-p4yiSoFI-xVzdf8LA,575
+chinadata-0.3.8.dist-info/METADATA,sha256=bKGFgc537irq0w9PH2N9AAUpOeRtwvZyAfTXzDEee6w,161
+chinadata-0.3.8.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+chinadata-0.3.8.dist-info/top_level.txt,sha256=Ptxvu9uzRHP4kVGRf413KuwSvsY2mk_KgQJ6ssWwisU,10
+chinadata-0.3.8.dist-info/RECORD,,

{chinadata-0.3.6.dist-info → chinadata-0.3.8.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.42.0)
+Generator: setuptools (75.6.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

chinadata-0.3.6.dist-info/RECORD

@@ -1,7 +0,0 @@
-chinadata/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-chinadata/ca_data.py,sha256=FlwtmAxyyYMAOCrMJ4dmrCfwjgbCQ9HIfHbQOVgN8Cg,37524
-chinadata/example.py,sha256=eKj2Hl5KVq9EJGWVY2ZoRJOqZ-p4yiSoFI-xVzdf8LA,575
-chinadata-0.3.6.dist-info/METADATA,sha256=57veZ5FXG1l93m2MkRViY3MGb6W8DwQ6CLaEh_-ZREw,163
-chinadata-0.3.6.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-chinadata-0.3.6.dist-info/top_level.txt,sha256=Ptxvu9uzRHP4kVGRf413KuwSvsY2mk_KgQJ6ssWwisU,10
-chinadata-0.3.6.dist-info/RECORD,,