From ed31938324700b7c4588883ddfa97633b9ce6618 Mon Sep 17 00:00:00 2001
From: 62bfaa38f5bc8d75ce8dc685 <62bfaa38f5bc8d75ce8dc685@devide>
Date: Fri, 5 May 2023 04:20:31 +0000
Subject: [PATCH] UPDATE

---
 .inscode   | 2 +-
 install.sh | 2 +-
 main.py    | 8 ++++----
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/.inscode b/.inscode
index 8ae4373..be42c1f 100644
--- a/.inscode
+++ b/.inscode
@@ -4,6 +4,6 @@ run = "bash install.sh"
 VIRTUAL_ENV = "/root/${PROJECT_DIR}/venv"
 PATH = "${VIRTUAL_ENV}/bin:${PATH}"
 PYTHONPATH = "$PYTHONHOME/lib/python3.10:${VIRTUAL_ENV}/lib/python3.10/site-packages"
-REPLIT_POETRY_PYPI_REPOSITORY = "http://mirrors.csdn.net.cn/repository/csdn-pypi-mirrors/simple"
+REPLIT_POETRY_PYPI_REPOSITORY = " https://pypi.tuna.tsinghua.edu.cn/simple "
 MPLBACKEND = "TkAgg"
 POETRY_CACHE_DIR = "/root/${PROJECT_DIR}/.cache/pypoetry"
\ No newline at end of file
diff --git a/install.sh b/install.sh
index 8ae1670..588e66f 100644
--- a/install.sh
+++ b/install.sh
@@ -2,7 +2,7 @@ wget 'https://dl.google.com/linux/direct/google-chrome-stable_current_amd64.deb'
 dpkg -i google-chrome-stable_current_amd64.deb
 apt --fix-broken install -y
 dpkg -i google-chrome-stable_current_amd64.deb
-pip install -r requirements.txt
+pip install -r requirements.txt -i  https://pypi.tuna.tsinghua.edu.cn/simple 
 
 mkdir -p /usr/share/fonts/chinese/        #创建中文字体目录
 cp simsun.ttf /usr/share/fonts/chinese/     #将字体文件拷贝到/usr/share/fonts/chinese/中
diff --git a/main.py b/main.py
index 24a6273..88f51d7 100644
--- a/main.py
+++ b/main.py
@@ -1,10 +1,10 @@
 from flask import Flask, render_template, request, send_from_directory
 from selenium import webdriver
 from flask import Flask, render_template, request, send_file, jsonify
-import os
+# import os
 app = Flask(__name__)
 app.config['UPLOAD_FOLDER'] = 'static/uploads/'
-import time
+# import time
 
 def take_screenshot(url):
     # 创建 Chrome 浏览器对象
@@ -19,9 +19,9 @@ def take_screenshot(url):
     driver.maximize_window()
     
     driver.get(url)
-    time.sleep(5)
+    # time.sleep(5)
     driver.implicitly_wait(10)
-    file_name = os.path.join(app.config['UPLOAD_FOLDER'], f'{os.urandom(8).hex()}.png')
+    file_name = 'static/uploads/screenshot.png'
     driver.save_screenshot(file_name)
     driver.quit()
     # # 裁剪图片
-- 
GitLab