GPT session账号导入

netflix-dao/src/main/java/com/cyksj/model/entity/ChatgptSession.java

@@ -88,5 +88,9 @@ public class ChatgptSession extends BaseEntity {
      */
     private String relateEmail;
 
-
+    /**
+     * 会员类型
+     * 普通、Plus、Team、Pro
+     */
+    private String type;
 }

netflix-dao/src/main/java/com/cyksj/model/excel/ExcelGptSessionAccountData.java

@@ -0,0 +1,72 @@
+package com.cyksj.model.excel;
+
+import com.alibaba.excel.annotation.ExcelProperty;
+import lombok.Getter;
+import lombok.Setter;
+
+/**
+ * 项目名: yhlxj
+ * 文件名: ExcelGptSessionAccountData
+ * 创建者: JavaZou
+ * 创建时间:2024/12/12 16:21
+ */
+@Getter
+@Setter
+public class ExcelGptSessionAccountData {
+	@ExcelProperty("账号")
+	private String account;
+
+	/**
+	 * 密码
+	 */
+	@ExcelProperty("密码")
+	private String password;
+
+
+	/**
+	 * 状态
+	 */
+	@ExcelProperty("状态")
+	private String statusStr;
+
+
+	/**
+	 * 账号类型
+	 *（单选:普通、Plus、Team、Pro)
+	 */
+	@ExcelProperty("账号类型（单选:普通、Plus、Team、Pro)")
+	private String type;
+
+	/**
+	 * 双重验证
+	 */
+	@ExcelProperty("双重验证")
+	private String apiSecret;
+
+	/**
+	 * 备用安全码
+	 */
+	@ExcelProperty("备用安全码")
+	private String verifyCodes;
+
+	/**
+	 * 邮箱密码
+	 */
+	@ExcelProperty("邮箱密码")
+	private String emailPassword;
+
+	@ExcelProperty("关联邮箱")
+	private String relateEmail;
+
+	@ExcelProperty("备注")
+	private String remark;
+
+//	@ExcelProperty("officialSession")
+//	private String officialSession;
+//
+//	@ExcelProperty("access_token")
+//	private String accessToken;
+//
+//	@ExcelProperty("refresh_token")
+//	private String refreshToken;
+}

netflix-dao/src/main/java/com/cyksj/model/excel/ExcelPrepareAccountData.java

@@ -13,7 +13,7 @@ public class ExcelPrepareAccountData {
     private String account;
 
     /**
-     * 省份
+     * 密码
      */
     @ExcelProperty("密码")
     private String password;

netflix-web/src/main/java/com/cyksj/web/controller/manage/account/CmsGptCarController.java

@@ -1,5 +1,6 @@
 package com.cyksj.web.controller.manage.account;
 
+import cn.hutool.core.bean.BeanUtil;
 import cn.hutool.core.date.DateUtil;
 import cn.hutool.core.util.ObjectUtil;
 import cn.hutool.core.util.RandomUtil;
@@ -8,6 +9,9 @@ import cn.hutool.http.HttpRequest;
 import cn.hutool.http.HttpUtil;
 import cn.hutool.json.JSONArray;
 import cn.hutool.json.JSONObject;
+import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.context.AnalysisContext;
+import com.alibaba.excel.event.AnalysisEventListener;
 import com.alibaba.excel.support.ExcelTypeEnum;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.cyksj.common.exception.BusinessRuntimeException;
@@ -21,6 +25,7 @@ import com.cyksj.model.dto.ChatgptChangeNodesDto;
 import com.cyksj.model.entity.ChatgptSession;
 import com.cyksj.model.entity.SysConfig;
 import com.cyksj.model.excel.ExcelGptMirrorAccountData;
+import com.cyksj.model.excel.ExcelGptSessionAccountData;
 import com.cyksj.model.views.ChatgptCarSessionView;
 import com.cyksj.redis.RedisService;
 import com.cyksj.service.chatgpt.ChatGptAccountService;
@@ -34,9 +39,11 @@ import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
 
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
 import java.util.List;
 import java.util.Optional;
 
@@ -301,4 +308,37 @@ public class CmsGptCarController {
 		}
 	}
 
+	/**
+	 * 下载GPT镜像车队模板
+	 */
+	@GetMapping("/export/gpt/session/template")
+	public void getGptSessionTemplate(HttpServletResponse response) {
+		EasyExcelUtils.createTemplateExcel(response, ExcelGptSessionAccountData.class, "导入GPT镜像车队账号模板", "导入GPT镜像车队账号模板", ExcelTypeEnum.XLSX, null);
+	}
+
+	/**
+	 * 上传GPT镜像车队
+	 */
+	@GetMapping("/import/gpt/session")
+	public Result<String> importGptSession(MultipartFile file) throws IOException {
+		EasyExcel.read(file.getInputStream(), ExcelGptSessionAccountData.class, new AnalysisEventListener<ExcelGptSessionAccountData>() {
+			@Override
+			public void invoke(ExcelGptSessionAccountData data, AnalysisContext analysisContext) {
+				ChatgptSession session = new ChatgptSession();
+				BeanUtil.copyProperties(data, session);
+				session.setStatus(0);
+				if ("有效".equals(data.getStatusStr())) {
+					session.setStatus(1);
+				}
+				//生成gptSession
+				postMirrorCarSession(session);
+			}
+			@Override
+			public void doAfterAllAnalysed(AnalysisContext analysisContext) {
+
+			}
+		}).sheet().doRead();
+		return GatewayResponse.SUCCESS.newBuilder().toResult();
+	}
+
 }