Spaces:
Sleeping
Sleeping
add get_cur_path
Browse files- app.py +13 -14
- requirements.txt +2 -1
- test_mysql.py +24 -0
app.py
CHANGED
@@ -3,19 +3,15 @@
|
|
3 |
from fastapi import FastAPI, HTTPException, Response, Depends
|
4 |
# from starlette.requests import Request
|
5 |
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
|
10 |
-
|
11 |
|
12 |
from dotenv import load_dotenv
|
13 |
import os
|
14 |
-
|
15 |
-
# del os.environ['DB_PATH']
|
16 |
-
# os.environ.pop('DB_PATH', None)
|
17 |
-
# for key, value in os.environ.items():
|
18 |
-
# print(f"{key}={value}")
|
19 |
|
20 |
# 加载.env文件
|
21 |
load_dotenv()
|
@@ -28,13 +24,12 @@ def get_db_path():
|
|
28 |
|
29 |
app = FastAPI()
|
30 |
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
|
35 |
@app.get("/")
|
36 |
def greet_json(db_path: str = Depends(get_db_path)):
|
37 |
-
return f"DB_PATH: {db_path}"
|
38 |
return {"Hello": "World!"}
|
39 |
@app.get("/init_db")
|
40 |
def init_db():
|
@@ -56,4 +51,8 @@ CREATE TABLE api_keys (
|
|
56 |
# """
|
57 |
# Db('api_keys.db').execute_query(sql)
|
58 |
|
59 |
-
# create_user_api_key('i_am_tanbushi', 'llm', '1', '0', 'dest_api_key')
|
|
|
|
|
|
|
|
|
|
3 |
from fastapi import FastAPI, HTTPException, Response, Depends
|
4 |
# from starlette.requests import Request
|
5 |
|
6 |
+
from routers.webtools_v1 import router as webtools_router
|
7 |
+
from routers.users_v1 import router as users_router
|
8 |
+
from routers.openai_v1 import router as openai_router
|
9 |
|
10 |
+
from db import Db
|
11 |
|
12 |
from dotenv import load_dotenv
|
13 |
import os
|
14 |
+
|
|
|
|
|
|
|
|
|
15 |
|
16 |
# 加载.env文件
|
17 |
load_dotenv()
|
|
|
24 |
|
25 |
app = FastAPI()
|
26 |
|
27 |
+
app.include_router(webtools_router, prefix="/airs/v1", tags=["webtools"])
|
28 |
+
app.include_router(users_router, prefix="/airs/v1", tags=["users"])
|
29 |
+
app.include_router(openai_router, prefix="/airs/v1", tags=["openai"])
|
30 |
|
31 |
@app.get("/")
|
32 |
def greet_json(db_path: str = Depends(get_db_path)):
|
|
|
33 |
return {"Hello": "World!"}
|
34 |
@app.get("/init_db")
|
35 |
def init_db():
|
|
|
51 |
# """
|
52 |
# Db('api_keys.db').execute_query(sql)
|
53 |
|
54 |
+
# create_user_api_key('i_am_tanbushi', 'llm', '1', '0', 'dest_api_key')
|
55 |
+
|
56 |
+
@app.get("/cur_path")
|
57 |
+
def cur_path():
|
58 |
+
return {"cur_path": os.getcwd()}
|
requirements.txt
CHANGED
@@ -4,4 +4,5 @@ captcha
|
|
4 |
python-dotenv
|
5 |
langchain_google_genai==1.0.10
|
6 |
langchain-openai==0.1.25 # 使用 openai 或兼容 openai
|
7 |
-
# sqlite-web
|
|
|
|
4 |
python-dotenv
|
5 |
langchain_google_genai==1.0.10
|
6 |
langchain-openai==0.1.25 # 使用 openai 或兼容 openai
|
7 |
+
# sqlite-web
|
8 |
+
mysql-connector-python
|
test_mysql.py
ADDED
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import mysql.connector
|
2 |
+
|
3 |
+
# 配置数据库连接参数
|
4 |
+
config = {
|
5 |
+
'user': 'tanbushi', # 替换为你的MySQL用户名
|
6 |
+
'password': 'Tangeqin1968', # 替换为你的MySQL密码
|
7 |
+
'host': 'db4free.net', # 数据库主机地址,本地默认是localhost
|
8 |
+
'database': 'apimapper', # 你要连接的数据库名
|
9 |
+
'raise_on_warnings': True
|
10 |
+
}
|
11 |
+
|
12 |
+
# 建立连接
|
13 |
+
try:
|
14 |
+
cnx = mysql.connector.connect(**config)
|
15 |
+
print("Connection established")
|
16 |
+
except mysql.connector.Error as err:
|
17 |
+
print("Error:", err)
|
18 |
+
|
19 |
+
cursor = cnx.cursor()
|
20 |
+
query = ("SELECT COUNT(*) FROM users") # 替换为你的表名和查询
|
21 |
+
cursor.execute(query)
|
22 |
+
count = cursor.fetchone()[0]
|
23 |
+
print(f"表中的记录数: {count}")
|
24 |
+
cursor.close()
|