From 00d969dfefc4f12a0ac702e2061dc5acc1501443 Mon Sep 17 00:00:00 2001
From: pzrxqba79 <1994211138@qq.com>
Date: Fri, 11 Oct 2024 14:27:10 +0800
Subject: [PATCH] sty.py

---
 sty.py | 81 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 81 insertions(+)
 create mode 100644 sty.py

diff --git a/sty.py b/sty.py
new file mode 100644
index 0000000..51d0a40
--- /dev/null
+++ b/sty.py
@@ -0,0 +1,81 @@
+const express = require('express');
+const multer = require('multer');
+const cors = require('cors');
+const fs = require('fs');
+const path = require('path');
+const morgan = require('morgan');
+
+const app = express();
+const port = 3000;
+
+// Middleware
+app.use(cors({
+    origin: 'http://your-frontend-domain.com', // 指定允许的源
+    methods: ['GET', 'POST'],
+}));
+app.use(express.json());
+app.use(express.static(path.join(__dirname, 'public'))); // 提供静态文件
+app.use(morgan('combined')); // 记录请求日志
+
+// 创建上传目录
+const uploadsDir = path.join(__dirname, 'uploads');
+if (!fs.existsSync(uploadsDir)) {
+    fs.mkdirSync(uploadsDir, { recursive: true });
+}
+
+// Multer setup for file uploads
+const storage = multer.diskStorage({
+    destination: (req, file, cb) => {
+        cb(null, uploadsDir); // 确保上传目录存在
+    },
+    filename: (req, file, cb) => {
+        cb(null, Date.now() + path.extname(file.originalname)); // Append timestamp to avoid name collisions
+    }
+});
+
+const upload = multer({
+    storage,
+    limits: {
+        fileSize: 5 * 1024 * 1024 // 限制文件大小为5MB
+    },
+    fileFilter: (req, file, cb) => {
+        const filetypes = /csv|txt|json/; // 允许的文件类型
+        const extname = filetypes.test(path.extname(file.originalname).toLowerCase());
+        const mimetype = filetypes.test(file.mimetype);
+        
+        if (mimetype && extname) {
+            return cb(null, true);
+        }
+        cb(new Error('Error: File type not supported!'));
+    }
+});
+
+// Endpoint to handle file import
+app.post('/import', upload.single('file'), (req, res) => {
+    if (!req.file) {
+        return res.status(400).send('No file uploaded.');
+    }
+    console.log(req.file);
+    res.send('File imported successfully!');
+});
+
+// Endpoint to handle file export
+app.get('/export', (req, res) => {
+    const filePath = path.join(__dirname, 'exports', 'sample-data.csv'); // 确保该文件存在
+    res.download(filePath, 'sample-data.csv', (err) => {
+        if (err) {
+            console.error(err);
+            res.status(500).send('Error exporting file.');
+        }
+    });
+});
+
+// 路由
+app.get('/api/hello', (req, res) => {
+    res.send('Hello, world!');
+});
+
+// 启动服务器
+app.listen(port, () => {
+    console.log(`Server running at http://localhost:${port}`);
+});
\ No newline at end of file