kaq-quant-common 0.1.96__tar.gz → 0.1.98__tar.gz

{kaq_quant_common-0.1.96 → kaq_quant_common-0.1.98}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kaq_quant_common
-Version: 0.1.96
+Version: 0.1.98
 Summary: 
 Author: kevinfuture
 Author-email: liuenbofuture@foxmail.com

{kaq_quant_common-0.1.96 → kaq_quant_common-0.1.98}/kaq_quant_common/resources/kaq_ddb_stream_write_resources.py

@@ -35,6 +35,19 @@ class KaqQuantDdbStreamWriteRepository:
             self.logger.error(f'KaqQuantDdbStreamWriteRepository.__init__ is occured error: {str(e)} - {str(traceback.format_exc())}')
         finally:
             mutex.release()
+    
+    def save_row(self, ddb_table_name: str, row:list=[]):
+        '''
+        调用此方法之前, 需要将dataframe中的字符串类型的值 ，添加引号
+        '''
+        if row is None or len(row) <= 0:
+            return
+        formatted_values = [f'\'{v}\'' if isinstance(v, str) else str(v) for v in row]
+        script = f"insert into {ddb_table_name} values({', '.join(str(x) for x in formatted_values)})"
+        try:
+            self.session.run(script, clearMemory=True)
+        except Exception as e:
+            self.logger.error(f'KaqQuantDdbStreamWriteRepository.save2stream is occured error: tableName is {ddb_table_name} - {str(e)} - {str(traceback.format_exc())}')
         
     def save2stream(self, ddb_table_name: str, df : pd.DataFrame):
         '''
@@ -230,6 +243,14 @@ class KaqQuantDdbStreamMTWWriteRepository:
         except Exception as e:
             self.logger.error(f'KaqQuantDdbStreamMTWWriteRepository.insert is occured error: {str(e)} - {str(traceback.format_exc())}')
             
+    def save2stream(self, row:list=[]):
+        try:
+            if row is None or len(row) <= 0:
+                return
+            self.batch_writer.insert(*row)
+        except Exception as e:
+            self.logger.error(f'KaqQuantDdbStreamMTWWriteRepository.insert is occured error: {str(e)} - {str(traceback.format_exc())}')
+            
     def stop(self):
         '''
         结束调用

{kaq_quant_common-0.1.96 → kaq_quant_common-0.1.98}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kaq_quant_common"
-version = "0.1.96"
+version = "0.1.98"
 description = ""
 authors = [
     {name = "kevinfuture",email = "liuenbofuture@foxmail.com"}