1.处理邮箱附件上传，发送邮件附件

api/api_v1/endpoints/interview.py

@@ -22,7 +22,7 @@ from schemas import ExtendendOAuth
 from utils.dingding import *
 from utils.jianli import get_resume
 from utils.re_to_jianli import fmt_txt, getText_pdf
-from utils.func import get_uid, send_affix_mail, location_file
+from utils.func import get_uid, send_affix_mail
 import crud, schemas
 from datetime import datetime
 from core.configuration import *
@@ -1415,6 +1415,49 @@ async def resume_affix(
         return schemas.Msg(code=400, msg='上传华为云失败', data='')
 
 
+@router.post("/mail_affix")
+async def mail_affix(
+        request: Request,
+        file: UploadFile = File(...),
+        ckdb: CKDrive = Depends(get_ck_db),
+        current_user: schemas.UserDB = Depends(deps.get_current_user)
+) -> schemas.Msg:
+    """邮箱附件上传到华为云"""
+    path_data = os.getcwd() + '/mail_affix'  # 当前文件所在的目录
+    if not os.path.exists(path_data):
+        os.makedirs(path_data)
+    contents = await file.read()
+    filename = file.filename
+    try:
+        with open(path_data + '/' + filename, "wb") as f:
+            #  将获取的file文件内容，写入到新文件中
+            f.write(contents)
+            f.close()
+    except:
+        return schemas.Msg(code=400, msg='上传文件有误', data='')
+    try:
+        # 附件正常上传到华为云
+        res = obsClient.putFile('legu-cdn-source', 'hrms/mail_affix/' + filename, path_data + '/' + filename)
+        if res.status < 300:
+            # 地址
+            url = res.body.objectUrl
+            resData = {'file_url': url, 'filename': 'hrms/mail_affix/' + filename}
+            res = json.dumps(resData)
+            # sql = f"""
+            # ALTER table HR.resumes update resume_affix_id = {url} WHERE uid = '{data_in.uid}'
+            # """
+            # await ckdb.execute(sql)
+            return schemas.Msg(code=200, msg='ok', data=res)
+        else:
+            # print('errorCode:', res.errorCode)
+            # print('errorMessage:', res.errorMessage)
+            return schemas.Msg(code=400, msg='上传华为云失败', data='')
+    except:
+        import traceback
+        print(traceback.format_exc())
+        return schemas.Msg(code=400, msg='上传华为云失败', data='')
+
+
 @router.post("/updata_entry")
 async def updata_entry(
         request: Request,
@@ -1459,44 +1502,33 @@ async def get_str_mail(
 @router.post("/send_str_mail")
 async def send_mail(
         request: Request,
-        #data_in: schemas.send_str_mail,
+        data_in: schemas.send_str_mail,
-        user_id: str =Form(...), # 接收者的唯一id
+        #file: List[UploadFile] = File(...),
-        name: str =Form(...), # 接收者的名字
-        email: str=Form(...),  # 接收者的邮箱
-        email_str: str =Form(...), # 需要发送的文本内容
-        file:Union[List[UploadFile], None] = File(None),
         db: AsyncIOMotorDatabase = Depends(get_database),
         current_user: schemas.UserDB = Depends(deps.get_current_user)
 ) -> schemas.Msg:
     """发送邮件"""
+    path_data = os.getcwd() + '/mail_affix'  # 当前文件所在的目录
+    if not os.path.exists(path_data):
+        os.makedirs(path_data)
+    filename = data_in.file.split('/')[-1]
     try:
-        if file == None:
+        res = obsClient.getObject('legu-cdn-source', data_in.file, path_data + '/' + filename)
-            # 发送邮件
+        if res.status < 300:
-            send_affix_mail(email_str, [email])
+            send_affix_mail(data_in.email_str, [data_in.email], path_data + '/' + filename)
-        else:
-            #带附件的
-            file_list=[]
-            for i in file:
-                contents = await i.read()
-                filename = i.filename
-                file_list.append(filename)
-                with open(location_file + '/' + filename, "wb") as f:
-                    #  将获取的file文件内容，写入到新文件中
-                    f.write(contents)
-            send_affix_mail(email_str, [email],file_list)
         # send_str_mail(data_in.email_str, data_in.email)
         # 保存发送邮件的记录
         now_time = str(datetime.now()).split('.')[0]
-        await crud.email_record.create(db, schemas.email_record(user_id=user_id, text=email_str,
+        await crud.email_record.create(db, schemas.email_record(user_id=data_in.user_id, text=data_in.email_str,
                                                                 times=now_time, name=name, state=1,
-                                                                read_status=0, type='email'))
+                                                                read_status=0, type='email',mail_affix=data_in.file))
         return schemas.Msg(code=200, msg='邮件发送成功', data='')
     except Exception as e:
         print(e)
         now_time = str(datetime.now()).split('.')[0]
-        await crud.email_record.create(db, schemas.email_record(user_id=user_id, text=email_str,
+        await crud.email_record.create(db, schemas.email_record(user_id=data_in.user_id, text=data_in.email_str,
                                                                 times=now_time, name=name, state=0,
-                                                                read_status=0, type='email'))
+                                                                read_status=0, type='email',mail_affix=data_in.file))
         return schemas.Msg(code=200, msg='邮件发送失败', data='')
 
 

schemas/email_record.py

@@ -11,6 +11,7 @@ class email_record(BaseModel):
     state: int  # 邮件是否发送成功
     read_status: int  # 是否已读
     type: str
+    mail_affix: str  # 邮件附件存储地
 
 
 class get_email_record(BaseModel):
@@ -25,3 +26,4 @@ class email_records(DBBase):
     state: int  # 邮件是否发送成功
     read_status: int  # 是否已读
     type: str  # 消息类型
+    mail_affix: str  # 邮件附件存储地

schemas/interview_plan.py

@@ -32,7 +32,6 @@ class Post_uid(BaseModel):
     uid: str  # 面试者简历在CK中的唯一标识
 
 
-
 class get_uids(BaseModel):
     uid: str  # 面试者简历在CK中的唯一标识
 
@@ -58,6 +57,7 @@ class send_str_mail(BaseModel):
     name: str  # 接收者的名字
     email: str  # 接收者的邮箱
     email_str: str  # 需要发送的文本内容
+    file: str  # 华为云的邮箱附件地址
 
 
 class user_id(BaseModel):
@@ -168,9 +168,6 @@ class nterview(BaseModel):
     # interview_round: int  # 面试轮次
 
 
-
-
-
 class interviews(BaseModel):
     date: dict  # 要查询的条件
 

utils/func.py

@@ -28,7 +28,7 @@ my_sender = '250213850@qq.com'  # 发件人邮箱账号
 my_pass = 'whrsugtgkstibjdj'  # 发件人邮箱密码
 subject = '入职通知'  # 邮件的主题，也可以说是标题
 mail_host = 'smtp.qq.com'
-location_file = 'C:/Users/Administrator/Desktop/ddemo'  # 默认读取转存的文件地址（上传至邮箱的文件）
+
 def get_uid():
     return hex(int(time.time() * 10 ** 7) + random.randint(0, 10000))[2:]
 
@@ -252,12 +252,12 @@ def send_str_mail(str_msg, my_user):
     server.quit()  # 关闭连接
 
 
-def send_affix_mail(str_msg: str, send_to: list, file_name: list = None):
+def send_affix_mail(str_msg: str, send_to: list, file_name: str = None):
     """
     发送邮件,包含文本或和附件
     :param str_msg: 发送的邮件内容
     :param send_to: 接收者的邮箱
-    :param file_name: 上传的附件
+    :param file_name: 上传的附件名，绝对路径
     """
     msg = MIMEMultipart()  # 创建一个带附件的实例
     msg["Subject"] = "面试通知"
@@ -268,9 +268,9 @@ def send_affix_mail(str_msg: str, send_to: list, file_name: list = None):
     msg.attach(part)
     # ---附件部分---
     if file_name != None:
-        for i in file_name:
+        part = MIMEApplication(open(file_name, 'rb').read())
-            part = MIMEApplication(open(location_file + '/' + i, 'rb').read())
+        filename=file_name.split('/')[-1]
-            part.add_header('Content-Disposition', 'attachment', filename=i)
+        part.add_header('Content-Disposition', 'attachment', filename=filename)
         msg.attach(part)
     smpt = smtplib.SMTP_SSL(mail_host, 465, 'utf-8')
     smpt.login(my_sender, my_pass)