Bladeren bron

取表结构后处理

liweiquan 2 jaren geleden
bovenliggende
commit
0d9a073ab7
1 gewijzigde bestanden met toevoegingen van 4 en 1 verwijderingen
  1. 4 1
      app/core/datasource/mysql.py

+ 4 - 1
app/core/datasource/mysql.py

@@ -102,6 +102,9 @@ class MysqlDS(DataSourceBase):
         return flat_map(lambda x: x, res[0])
 
     def get_table_schema(self, table_name):
+        def handle_col(x):
+            line = list(map(lambda s:  s.decode('utf-8') if type(s) == type(b'bytes') else  str(s), x))
+            return [':'.join(line[:3])]
         sql = f'describe `{self.database_name}`.{table_name}'
         logger.info(sql)
         res = self._execute_sql([sql])
@@ -109,7 +112,7 @@ class MysqlDS(DataSourceBase):
             res = [[str(i) , *x]for i, x in enumerate(res[0])]
             logger.info(res)
 
-            return flat_map(lambda x: [':'.join(x[:3])], res)
+            return flat_map(lambda x: handle_col(x), res)
         else:
             raise Exception('table not found')