增加数据库连接

weather/dao/models.py

+# coding: utf-8
+from sqlalchemy import Column, Integer, String, text
+from sqlalchemy.ext.declarative import declarative_base
+
+
+Base = declarative_base()
+metadata = Base.metadata
+
+
+class UserInfo(Base):
+    __tablename__ = 'user_info'
+
+    id = Column(Integer, primary_key=True, nullable=False)
+    nickname = Column(String(50), primary_key=True, nullable=False, server_default=text("''"))

weather/dao/sqlengin.py

+import sqlalchemy
+from sqlalchemy import create_engine
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy import Column, Integer, String
+from sqlalchemy.orm import sessionmaker
+
+
+engine = create_engine("mysql+pymysql://root:root@localhost/daylife?charset=utf8mb4", echo=True)
+# 创建DBSession类型:
+DBSession = sessionmaker(bind=engine)
+
+def getSession():
+    return DBSession()
+
+

weather/service.py

@@ -3,6 +3,8 @@ from flask import request
 import requests
 import json
 from common import BaseRes
+from weather.dao import sqlengin
+from weather.dao import models
 
 weather = Blueprint('weather', __name__, url_prefix='/weather')
 
@@ -21,4 +23,14 @@ def hello_world():
         data = response['data']
 
         return json.dumps(BaseRes(data['forecast']), default=lambda obj: obj.__dict__)
-    return json.dumps(BaseRes(status=404, message='无该城市数据，请检查参数是否正确！'), default=lambda obj: obj.__dict__)
\ No newline at end of file
+    return json.dumps(BaseRes(status=404, message='无该城市数据，请检查参数是否正确！'), default=lambda obj: obj.__dict__)
+
+@weather.route('/insert_user_info')
+def insert_user_info():
+    nickname = request.args.get('nickname')
+    session = sqlengin.getSession()
+    for i in range(0, 10000):
+        session.add(models.UserInfo(nickname=nickname))
+    session.commit()
+    session.close()
+    return '操作成功！'