Flask自定义命令

Flask自定义命令

一、老版Flask自定义命令

Flask 1.x 和 2.x 版本

在Flask的老版本中,可以通过 flask-script 扩展来添加自定义命令。

  1. 安装所需库:

    python 复制代码
    pip3 install Flask-Script==2.0.3
    pip3 install flask==1.1.4
    pip3 install markupsafe=1.1.1
  2. 在 Flask 应用中创建一个 Manager 对象,并使用装饰器定义自定义命令:

    python 复制代码
    from flask_script import Manager
    manager = Manager(app)
    
    @manager.command
    def custom(arg):
        print(arg)
  3. 运行自定义命令:

    python 复制代码
    python manage.py custom 123

二、Flask 新版本(2.x 及以后)

Flask 新版本通过内置的 click 模块支持自定义命令,不再需要 flask-script

  1. 使用 click 装饰器定义命令:

    python 复制代码
    from flask import Flask
    import click
    
    app = Flask(__name__)
    
    @app.cli.command("create-user")
    @click.argument("name")
    def create_user(name):
        print(name)
  2. 运行自定义命令:

    python 复制代码
    flask --app 文件名:app create-user bruce

    如果你的脚本文件名为 app.py,则可以简化命令为:

    python 复制代码
    flask create-user lqz

三、flask自定义命令示例

  • 要求:制定一个flask命令,导入一个excel,把数据同步到 user表中

  • 代码示例

    • python 复制代码
      from flask import Flask, current_app
      import click
      import pymysql
      from openpyxl import load_workbook
      
      app = Flask(__name__)
      
      @app.cli.command("import_excel")
      @click.argument("excel_path")
      def import_excel(excel_path):
       """将指定路径的 Excel 文件数据导入到 user 表中"""
       # 加载 Excel 文件
       workbook = load_workbook(filename=excel_path)
       sheet = workbook.active
      
       # 连接数据库
       connection = pymysql.connect(host='localhost',
                                    port=3306,
                                    user='root',
                                    password='000',
                                    database='test',
                                    cursorclass=pymysql.cursors.DictCursor)
      
       try:
           with connection.cursor() as cursor:
               for row in sheet.iter_rows(min_row=2, values_only=True):  # 假设第一行为表头
                   # 构造插入语句,假设 Excel 文件中有 'user'
                   sql = "INSERT INTO user (name) VALUES (%s)"
                   cursor.execute(sql, row)
               connection.commit()
       finally:
           connection.close()
      
       print(f"Excel 数据导入完成,文件:{excel_path}")
      
      
      if __name__ == '__main__':
       app.run()

四、Django自定义命令

Django允许在应用中创建自定义命令,步骤如下:

  1. 在应用目录下创建 management/commands 文件夹结构。

  2. commands 文件夹中创建一个新的 Python 文件,该文件的名字将是你的命令名称。

  3. 在该文件中编写自定义命令的代码,继承 BaseCommand 类:

    python 复制代码
    from django.core.management.base import BaseCommand
    
    class Command(BaseCommand):
        help = '命令提示'
    
        def add_arguments(self, parser):
            parser.add_argument('path', nargs='*', type=str)
    
        def handle(self, *args, **kwargs):
            print('开始导入')
            print(args)
            print(kwargs)
  4. 使用自定义命令:

    python 复制代码
    python manage.py 文件名
相关推荐
齐 飞7 分钟前
MongoDB笔记01-概念与安装
前端·数据库·笔记·后端·mongodb
云空7 分钟前
《Python 与 SQLite:强大的数据库组合》
数据库·python·sqlite
LunarCod23 分钟前
WorkFlow源码剖析——Communicator之TCPServer(中)
后端·workflow·c/c++·网络框架·源码剖析·高性能高并发
神仙别闹24 分钟前
基于tensorflow和flask的本地图片库web图片搜索引擎
前端·flask·tensorflow
凤枭香1 小时前
Python OpenCV 傅里叶变换
开发语言·图像处理·python·opencv
码农派大星。1 小时前
Spring Boot 配置文件
java·spring boot·后端
测试杂货铺1 小时前
外包干了2年,快要废了。。
自动化测试·软件测试·python·功能测试·测试工具·面试·职场和发展
艾派森1 小时前
大数据分析案例-基于随机森林算法的智能手机价格预测模型
人工智能·python·随机森林·机器学习·数据挖掘
小码的头发丝、1 小时前
Django中ListView 和 DetailView类的区别
数据库·python·django
杜杜的man2 小时前
【go从零单排】go中的结构体struct和method
开发语言·后端·golang