vibora 经验

pip  install  vibora  tortoise-orm

from  tortoise  import  fields,  models
from  tortoise.contrib.pydantic  import  pydantic_model_creator

class  User(models.Model):
      id  =  fields.IntField(pk=True)
      username  =  fields.CharField(max_length=255,  unique=True)
      email  =  fields.CharField(max_length=255,  unique=True)

#  创建  Pydantic  model，方便在请求处理中使用
UserPydantic  =  pydantic_model_creator(User,  name='User')

from  tortoise  import  Tortoise

async  def  init_db():
      #  这里配置数据库连接
      TORTOISE_ORM  =  {
          'connections':  {
              'default':  {
                  'engine':  'asyncpg',
                  'host':  '127.0.0.1',
                  'port':  '5432',
                  'user':  'user',
                  'password':  'password',
                  'database':  'test_db',
              },
          },
          'apps':  {
              'models':  {
                  'models':  ['models'],    #  注册我们的models模块
                  'default_connection':  'default',
              },
          },
      }
     
      await  Tortoise.init(TORTOISE_ORM)
      await  Tortoise.generate_schemas()

from  vibora  import  Vibora,  Request
from  pydantic  import  BaseModel
from  tortoise.exceptions  import  DoesNotExist
from  models  import  User,  UserPydantic
from  db_config  import  init_db

app  =  Vibora()

@app.listener('before_server_start')
async  def  setup_db(app,  loop):
      await  init_db()

#  创建一个请求模型来验证输入
class  CreateUser(BaseModel):
      username:  str
      email:  str

@app.route('/users',  methods=['POST'])
async  def  create_user(request:  Request,  user:  CreateUser):
      user  =  await  User.create(username=user.username,  email=user.email)
      return  {'id':  user.id,  'username':  user.username,  'email':  user.email}

@app.route('/users/{user_id}',  methods=['GET'])
async  def  get_user(request:  Request,  user_id:  int):
      try:
          user  =  await  User.get(id=user_id).prefetch_related('user')
          return  {'id':  user.id,  'username':  user.username,  'email':  user.email}
      except  DoesNotExist:
          return  {'error':  'User  not  found'},  404

@app.route('/users/{user_id}',  methods=['PUT'])
async  def  update_user(request:  Request,  user_id:  int,  user:  CreateUser):
      try:
          await  User.filter(id=user_id).update(username=user.username,  email=user.email)
          return  {'message':  'User  updated  successfully'}
      except  DoesNotExist:
          return  {'error':  'User  not  found'},  404

@app.route('/users/{user_id}',  methods=['DELETE'])
async  def  delete_user(request:  Request,  user_id:  int):
      try:
          await  User.get(id=user_id).delete()
          return  {'message':  'User  deleted  successfully'}
      except  DoesNotExist:
          return  {'error':  'User  not  found'},  404

if  __name__  ==  '__main__':
      app.run(debug=True)

from  vibora  import  Vibora
from  db_config  import  init_db
from  routes  import  setup_routes

app  =  Vibora()

@app.listener('before_server_start')
async  def  setup_db(app,  loop):
      await  init_db()

#  设置路由
setup_routes(app)

if  __name__  ==  '__main__':
      app.run(debug=True)

from  pydantic  import  BaseModel
from  tortoise.exceptions  import  DoesNotExist
from  models  import  User,  UserPydantic
from  vibora  import  Request,  route

class  CreateUser(BaseModel):
      username:  str
      email:  str

@route('/users',  methods=['POST'])
async  def  create_user(request:  Request,  user:  CreateUser):
      user  =  await  User.create(username=user.username,  email=user.email)
      return  {'id':  user.id,  'username':  user.username,  'email':  user.email}

@route('/users/{user_id}',  methods=['GET'])
async  def  get_user(request:  Request,  user_id:  int):
      try:
          user  =  await  User.get(id=user_id).prefetch_related('user')
          return  {'id':  user.id,  'username':  user.username,  'email':  user.email}
      except  DoesNotExist:
          return  {'error':  'User  not  found'},  404

@route('/users/{user_id}',  methods=['PUT'])
async  def  update_user(request:  Request,  user_id:  int,  user:  CreateUser):
      try:
          await  User.filter(id=user_id).update(username=user.username,  email=user.email)
          return  {'message':  'User  updated  successfully'}
      except  DoesNotExist:
          return  {'error':  'User  not  found'},  404

@route('/users/{user_id}',  methods=['DELETE'])
async  def  delete_user(request:  Request,  user_id:  int):
      try:
          await  User.get(id=user_id).delete()
          return  {'message':  'User  deleted  successfully'}
      except  DoesNotExist:
          return  {'error':  'User  not  found'},  404

def  setup_routes(app):
      app.add_route(create_user)
      app.add_route(get_user)
      app.add_route(update_user)
      app.add_route(delete_user)

from  vibora  import  Vibora,  Request,  Response
from  vibora.responses  import  JSONResponse
from  db_config  import  init_db
from  routes  import  setup_routes
from  middleware  import  token_middleware
import  asyncio
import  datetime

app  =  Vibora()

@app.listener('before_server_start')
async  def  setup_db(app,  loop):
      await  init_db()

#  设置路由
setup_routes(app)

#  定时事务
async  def  schedule_transactions(app):
      while  True:
          #  第一个事务：每6秒处理一次
          await  asyncio.sleep(6)
          print("Running  the  first  transaction...")

          #  第二个事务：每33分钟处理一次
          await  asyncio.sleep(33  *  60)
          print("Running  the  second  transaction...")

          #  第三个事务：每天早上1点到5点之间处理
          now  =  datetime.datetime.now()
          if  now.hour  >=  1  and  now.hour  <  5:
              print("Running  the  third  transaction...")
              #  确保不会在1小时内多次执行
              await  asyncio.sleep((5  -  now.hour)  *  3600  -  now.minute  *  60)

#  跨域处理中间件
@app.middleware
async  def  cors_middleware(request:  Request,  handler):
      response  =  await  handler(request)
      response.headers['Access-Control-Allow-Origin']  =  '*'    #  允许所有来源
      response.headers['Access-Control-Allow-Methods']  =  'GET,  POST,  PUT,  DELETE,  OPTIONS'
      response.headers['Access-Control-Allow-Headers']  =  'Content-Type,  Authorization'
      return  response

#  使用中间件验证token
app.middleware(token_middleware)

#  启动定时事务
@app.listener('after_server_start')
async  def  start_scheduling(app,  loop):
      loop.create_task(schedule_transactions(app))

if  __name__  ==  '__main__':
      app.run(debug=True)

from  vibora  import  Request,  Response
from  typing  import  Awaitable
from  jose  import  jwt    #  假设使用的是PyJWT库

async  def  token_middleware(request:  Request,  handler:  Awaitable)  ->  Response:
      token  =  request.headers.get('Authorization',  '').replace('Bearer  ',  '')
      if  token:
          try:
              #  假设你的secret_key是在环境变量中
              user_info  =  jwt.decode(token,  'secret_key',  algorithms=['HS256'])
              #  将用户信息添加到请求中
              request.state.user_info  =  user_info
          except:
              return  JSONResponse({'error':  'Invalid  token'},  status_code=401)
      else:
          return  JSONResponse({'error':  'Token  is  missing'},  status_code=401)
      return  await  handler(request)