tanbushi commited on
Commit
e77218b
·
1 Parent(s): d6b9283

add get_cur_path

Browse files
Files changed (3) hide show
  1. app.py +13 -14
  2. requirements.txt +2 -1
  3. 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
- # 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
- # if 'DB_PATH' in os.environ:
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
- # app.include_router(webtools_router, prefix="/airs/v1", tags=["webtools"])
32
- # app.include_router(users_router, prefix="/airs/v1", tags=["users"])
33
- # app.include_router(openai_router, prefix="/airs/v1", tags=["openai"])
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()