/
app.py
71 lines (54 loc) · 1.54 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
from flask import Flask, render_template, url_for
from database import db
from models import User, Post, Comment
from faker import Faker
import random
app = Flask(__name__)
app.config["SQLALCHEMY_DATABASE_URI"] = 'sqlite:///example.sqlite3'
app.config["SQLALCHEMY_ECHO"] = False
app.secret_key = 'SSHasdasd'
db.init_app(app)
@app.cli.command("start")
def start():
fake = Faker()
db.drop_all()
db.create_all()
# Users
for _ in range(100):
u = User()
u.username = fake.unique.user_name()
u.password = fake.password()
u.email = fake.email()
db.session.add(u)
db.session.commit()
# Posts
all_users = User.query.all()
for _ in range(1000):
p = Post()
p.user = random.choice(all_users)
p.title = fake.sentence()
p.content = fake.paragraph()
db.session.add(p)
db.session.commit()
# Comments
all_posts = Post.query.all()
for _ in range(5000):
c = Comment()
c.content = fake.paragraph()
c.user = random.choice(all_users)
c.post = random.choice(all_posts)
db.session.add(c)
db.session.commit()
@app.errorhandler(404)
def error_page(id):
return render_template('404.html'), 404
@app.route('/')
def home():
p = Post.query.order_by(Post.dttm.desc()).all()
return render_template('home.html', posts=p)
@app.route('/post/<int:id>')
def post(id):
p = Post.query.filter_by(id=id).first_or_404()
return render_template('post.html', post=p)
if __name__ == '__main__':
app.run()