1

fastapi-demo/util/upload_file_util.py

@@ -0,0 +1,8 @@
+from fastapi import UploadFile
+
+
+def save_upload_file(file: UploadFile, file_name=None):
+    if not file_name:
+        file_name = file.filename
+    with open(file_name, 'wb') as f:
+        f.write(file.file.read())

tmp/green_plum.py

@@ -1,12 +1,16 @@
 import pandas as pd
 import psycopg2
 
+
 def queryGP(sql):
-    ## 连接到一个给定的数据库
-    conn = psycopg2.connect(dbname="cdpqa", user="cdpqa", password="Cdpqa@123456",port="5432", host="eip-greenplum-2.qa.mlamp.cn", client_encoding="UTF-8")
+    # 连接到一个给定的数据库
+    conn = psycopg2.connect(dbname="cdpqa", user="cdpqa", password="Cdpqa@123456", port="5432",
+                            host="eip-greenplum-2.qa.mlamp.cn", client_encoding="UTF-8")
     df = pd.read_sql(sql, con=conn)
-    ## 关闭数据库连接
+    # 关闭数据库连接
     conn.close()
     return df
+
+
 df = queryGP("select * from pg_tables")
 print(df)