Parcourir la source

判断kerberos文件夹是否存在

liweiquan il y a 2 ans
Parent
commit
2ec098373e
2 fichiers modifiés avec 3 ajouts et 1 suppressions
  1. 3 0
      app/utils/get_kerberos.py
  2. 0 1
      server.py

+ 3 - 0
app/utils/get_kerberos.py

@@ -5,6 +5,9 @@ from app.common.minio import minio_client
 
 def get_kerberos_to_local(uri: str):
     file_name = uri.split("/")[-1]
+    kerberos_exists = os.path.exists('./assets/kerberos')
+    if not kerberos_exists:
+        os.makedirs('./assets/kerberos')
     f_exists = os.path.exists('./assets/kerberos/'+file_name)
     if not f_exists:
         res = minio_client.get_file(uri)

+ 0 - 1
server.py

@@ -20,7 +20,6 @@ from app.utils.get_kerberos import get_kerberos_to_local
 from utils.sx_log import format_print
 
 format_print()
-
 Base.metadata.create_all(bind=engine)
 app = FastAPI(docs_url='/jpt/docs', redoc_url='/jpt/redoc', title="DAG In Airflow管理系统")