First commit

Author: alphaProgrammingNow

.flaskenv

@@ -0,0 +1,2 @@
+FLASK_ENV=development
+FLASK_APP=watchlist

requirements.txt

@@ -0,0 +1,11 @@
+Click==7.0
+coverage==5.1
+Flask==1.1.1
+Flask-Login==0.4.1
+Flask-SQLAlchemy==2.4.0
+itsdangerous==1.1.0
+Jinja2==2.10.1
+MarkupSafe==1.1.1
+python-dotenv==0.10.3
+SQLAlchemy==1.3.6
+Werkzeug==0.15.4

watchlist/__init__.py

@@ -0,0 +1,74 @@
+import sys
+import os
+from flask import Flask
+from flask_sqlalchemy import SQLAlchemy
+from flask_login import LoginManager
+
+
+# SQLite URI compatible
+WIN = sys.platform.startswith('win')
+if WIN:
+    prefix = 'sqlite:///'
+else:
+    prefix = 'sqlite:////'
+
+
+
+
+app = Flask(__name__)
+# os.getenv('SECRET_KEY', 'dev') 表示读取系统环境变量 SECRET_KEY 的值，如果没有获取到，则使用 dev
+app.config['SECRET_KEY'] = os.getenv('SECRET_KEY', 'dev')
+# 注意更新这里的路径，把 app.root_path 添加到 os.path.dirname() 中
+# 以便把文件定位到项目根目录
+app.config['SQLALCHEMY_DATABASE_URI'] = prefix + os.path.join(os.path.dirname(app.root_path),os.getenv('DATABASE_FILE', 'data.db'))
+app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
+
+db = SQLAlchemy(app)
+login_manager = LoginManager(app)
+
+# 除了实例化扩展类之外，我们还要实现一个“用户加载回调函数”
+@login_manager.user_loader
+def load_user(user_id): # 创建用户加载回调函数，接受用户 ID 作为参数
+    from watchlist.models import User
+    user = User.query.get(int(user_id)) # 用 ID 作为 User 模型的主键查询对应的用户
+    return user # 返回用户对象
+    
+ 
+ 
+# 未登录用户访问delete等页面时，我们不能让他访问，我们要让他重定向，重定向到哪里？就在这里决定
+# 登录视图端点（函数名）
+login_manager.login_view='login'
+
+
+
+@app.context_processor
+def inject_user():
+    from watchlist.models import User
+    user = User.query.first()
+    return dict(user=user)
+# 这个函数返回的变量（以字典键值对的形式）将会统一注入到每一个模板的上下文环境中，因此可以直接在模板中使用。
+
+
+from watchlist import views,errors,commands
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+

watchlist/__pycache__/__init__.cpython-39.pyc

@@ -0,0 +1,65 @@
+import click
+
+from watchlist import app,db
+from watchlist.models import User,Movie
+
+@app.cli.command()
+@click.option('--drop',is_flag=True,help='Create after drop')
+def initdb(drop):
+    """Initialize the database"""
+    if drop: # 判断是否输入了选项
+        db.drop_all()
+    db.create_all()
+    click.echo('Initialize database.') #输出提示信息
+
+
+@app.cli.command()
+def forge():
+    """Generate fake data"""
+    # 如果没有db.drop_all(),那么,就在数据库原有的两张表的基础上，继续添加记录
+    db.create_all()
+    name = 'alpha2'
+    movies = [
+        {'title': 'My Neighbor Totoro', 'year': '1988'},
+        {'title': 'Dead Poets Society', 'year': '1989'},
+        {'title': 'A Perfect World', 'year': '1993'},
+        {'title': 'Leon', 'year': '1994'},
+        {'title': 'Mahjong', 'year': '1996'},
+        {'title': 'Swallowtail Butterfly', 'year': '1996'},
+        {'title': 'King of Comedy', 'year': '1999'},
+        {'title': 'Devils on the Doorstep', 'year': '1999'},
+        {'title': 'WALL-E', 'year': '2008'},
+        {'title': 'The Pork of Music', 'year': '2012'},
+        {'title': 'Harry Potter', 'year': '2012'},
+        {'title': '我和我的家乡', 'year': '2020'},
+    ]
+
+    user = User(name=name)
+    db.session.add(user)
+    for m in movies:
+        movie = Movie(title=m['title'],year=m['year'])
+        db.session.add(movie)
+    db.session.commit()
+    click.echo('Done.')
+
+
+
+@app.cli.command()#注册为命令
+@click.option('--userName',prompt=True,help='login user name')
+@click.option('--password',prompt=True,hide_input=True,confirmation_prompt=True,help='login user password')
+def admin(username,password):
+    db.create_all()
+    user = User.query.first()
+    if user is not None:
+        click.echo('更新用户名...')
+        user.username = username
+        user.set_password(password)
+
+    else:
+        click.echo('创建新用户...')
+        user = User(username=username,name='Admin')
+        user.set_password(password)
+        db.session.add(user)
+
+    db.session.commit()
+    click.echo('Done.')

watchlist/__pycache__/commands.cpython-39.pyc

@@ -0,0 +1,23 @@
+from flask import render_template
+from watchlist import app
+
+# 使用 app.errorhandler() 装饰器注册一个错误处理函数
+@app.errorhandler(404) # 传入要处理的错误代码
+def page_not_found(e): # 接受异常对象作为参数
+    # user = User.query.first()
+    return render_template('errors/404.html'),404  # 返回模板和状态码
+# 普通的视图函数之所以不用写出状态码，是因为默认会使用 200 状态码，表示成功
+
+
+@app.errorhandler(400)
+def bad_request(e):
+    return render_template('errors/400.html'), 400
+
+
+@app.errorhandler(500)
+def internal_server_error(e):
+    return render_template('errors/500.html'), 500
+
+
+
+