first push

utils/updateObs.py

@@ -0,0 +1,71 @@
+from obs import ObsClient
+from datetime import datetime
+#from config import cfg
+from utils.config import cfg
+import os,threading
+import time as ti
+import base64,requests,cv2,shutil
+lock = threading.Lock()
+obsClient = ObsClient(
+    access_key_id='LHXJC7GIC2NNUUHHTNVI',
+    secret_access_key='sVWvEItrFKWPp5DxeMvX8jLFU69iXPpzkjuMX3iM',
+    server='https://obs.cn-east-3.myhuaweicloud.com'
+)
+bucketName = 'ieemoo-ai'
+
+def AddObs(file_path, status):
+    with lock:
+        if not cfg.flag:
+            addobs(file_path, status)
+        else:
+            if status == '02':
+                addobs(file_path, status)
+            else:
+                objectkey = os.path.basename(file_path)
+                f_dist = os.sep.join([cfg.tempvideos, status+'_'+objectkey])
+                shutil.move(file_path, f_dst)
+
+def addobs(file_path, status): #save videos
+    ti.sleep(5)
+    videoUuid = os.path.basename(file_path).split('_')[1]
+    json_data = {'videoUuid': videoUuid}
+    resp = requests.post(url=cfg.Vre,
+                        data=json_data)
+    status = resp.json()
+    try:
+        status = status['data']
+    except Exception as e:
+        ti.sleep(5)
+        resp = requests.post(url=cfg.Vre,data=json_data)
+        status = resp.json()
+        status = status['data']
+
+    objectkey = os.path.basename(file_path)
+    time = os.path.basename(file_path).split('-')[0]
+    if objectkey.split('.')[-1] in ['avi','mp4']:
+        objectkey = 'videos/'+time+'/'+status+'/'+status+'_'+objectkey
+        resp = obsClient.putFile(bucketName, objectkey, file_path)
+    os.remove(file_path)
+
+def Addimg(uuid_barcode):
+    time = uuid_barcode.split('-')[0].split('_')[-1]
+    objectkey = 'imgs/'+time+'/'+uuid_barcode+'.jpg'
+    file_path = os.sep.join([cfg.Tempimg, '5_'+uuid_barcode+'.jpg'])
+    if not os.path.exists(file_path):
+        file_path = os.sep.join([cfg.Tempimg, '3_'+uuid_barcode+'.jpg'])
+    if not os.path.exists(file_path):
+        file_path = os.sep.join([cfg.Tempimg, 'ex_'+uuid_barcode+'.jpg'])
+    resp = obsClient.putFile(bucketName, objectkey, file_path)
+
+def Addimg_content(uuid_barcode, context):
+    success, encoded_image = cv2.imencode(".jpg",context)
+    context = encoded_image.tobytes()
+    time = uuid_barcode.split('-')[0]
+    objectkey = 'imgs/'+time+'/'+uuid_barcode+'.jpg'
+    resp = obsClient.putContent(bucketName, objectkey, context)
+
+if __name__ == '__main__':
+    import cv2
+    context = cv2.imread('/home/lc/project/ieemoo-ai-search/data/imgs/20230625-094651_37dd99b0-520d-457b-8615-efdb7f53b5b4_6907992825762.jpg')
+    uuid_barcode = '20230625-094651_37dd99b0-520d-457b-8615-efdb7f53b5b4'
+    Addimg_content(uuid_barcode, context)