def post(self, quote_id):
author = self.get_argument('author', '')
country = self.get_argument('country', '')
content = self.get_argument('content', '')
session = Session()
session.query(Quotes).filter_by(id=int(quote_id)).update({Quotes.author: author, Quotes.country: country,Quotes.quotes_content: content})
session.commit()
self.redirect('/quotes')
def post(self, user_id):
username = self.get_argument('username', '')
password = self.get_argument('password', '')
email = self.get_argument('email', '')
gender = self.get_argument('gender', '')
age = self.get_argument('age', ' ')
power = self.get_argument('power', '')
session =Session()
session.query(User).filter_by(id=int(user_id)).update({User.username: username, User.password: password, User.emial: email, User.gender: gender, User.age: age,User.power: int(power)})
session.commit()
self.redirect('/administrator')
def post(self):
username = self.get_argument('username', '')
username1 = self.get_argument('username1', '')
password = self.get_argument('password', '')
email = self.get_argument('email', '')
gender = self.get_argument('gender', '')
age = self.get_argument('age', ' ')
session = Session()
dict_data = {User.username:username1,User.password:password, User.emial:email,User.gender:gender,User.age:age}
session.query(User).filter_by(username=username).update(dict_data)
session.commit()
self.redirect('/')
def get(self, quote_id):
session = Session()
rows = session.query(Quotes).filter_by(id=quote_id).first()
session.delete(rows)
session.commit()
session.close()
self.redirect('/quotes')
def register():
if current_user.is_authenticated:
return redirect(url_for("home"))
if request.method == "GET":
return render_template("auth/register.html", user=current_user)
username = request.form.get("username")
password = request.form.get("password")
if not (username and len(username) >= 3 and password
and len(password) >= 4):
return render_template(
"auth/register.html",
error_text="Username has to be at least 3 symbols and pass min 5",
user=current_user,
)
if Session.query(User).filter_by(username=username).count():
return render_template(
"auth/register.html",
error_text=f"Username {username!r} already exists!",
user=current_user,
)
user = User(username, password)
Session.add(user)
try:
Session.commit()
except Exception as e:
logger.exception("Error creating user!")
raise InternalServerError(f"Could not create new user! Error: {e}")
login_user(user)
return redirect(url_for("home"))
def get(self, user_id):
session = Session()
rows = session.query(User).filter_by(id=user_id).first()
session.delete(rows)
session.commit()
session.close()
self.redirect('/administrator')
def login():
form = LoginForm()
if request.method == "POST" and form.validate():
user = Session.query(User).filter_by(
username=form.username.data).first()
login_user(user)
return redirect(url_for("main"))
return render_template("auth/login.html", form=form)
def get_all_posts(username=None):
'''
查询获取所有图片或者是特定的用户
:param username:如果没做，就是获取全部图片
:return:
'''
session = Session()
if username:
user = session.query(User).filter_by(username=username).first()
posts = session.query(Post).filter_by(user=user).all()
else:
posts = session.query(Post).all()
if posts:
return posts
else:
return []
def add_post(img_url, thumb_url, username):
session = Session()
user = session.query(User).filter_by(username=username).first()
post = Post(image_url=img_url, thumb_url=thumb_url, user_id=user.id)
session.add(post)
session.commit()
post_id = post.id
session.close()
return post_id
def update_post(post_id):
post = Session.query(Post).filter_by(id=post_id).first()
form = PostForm(obj=post)
if request.method == "POST" and form.validate():
post.title = request.form.get('title')
post.text = request.form.get('text')
post.image_url = request.form.get('image_url')
Session.commit()
return redirect(url_for("main"))
return render_template("posts/update.html", form=form, post_id=post.id)
def login():
if current_user.is_authenticated:
return redirect(url_for("index.html"))
if request.method == "GET":
return render_template("login.html")
username, password = get_username_and_password_from_form(request.form)
user = Session.query(User).filter_by(username=username).one_or_none()
if not user:
return render_template("login.html", error_text="User not found")
if user.password != User.hash_password(password):
return render_template("login.html",
error_text="Invalid username or password!")
login_user(user)
print("Uid:", user.id)
return redirect(url_for("index"))
def register(username, password1, password2):
ret = {'msg': 'other error', 'user_id': None}
if username and password1 and password2:
if password1 == password2:
session = Session()
user = session.query(User).filter_by(username=username).all()
if user:
msg = 'username is exists'
else:
new_user = User(username=username, password=hash(password1))
session.add(new_user)
session.commit()
ret['user_id'] = new_user.id
session.close()
msg = 'ok'
else:
msg = 'password1 != password2'
else:
msg = 'username or password is empty'
ret['msg'] = msg
return ret
def login():
if current_user.is_authenticated:
return redirect(url_for("home"))
if request.method == "GET":
return render_template("auth/login.html", user=current_user)
username = request.form.get("username")
password = request.form.get("password")
user = Session.query(User).filter_by(username=username).one_or_none()
if not user:
return render_template("auth/login.html", error_text="User not found")
if user.password != User.hash_password(password):
return render_template(
"auth/login.html",
error_text="Invalid username or password!",
user=current_user,
)
login_user(user)
return redirect(url_for("home"))
def is_exists(cls, username, password):
session = Session()
ret = session.query(exists().where(
and_(User.name == username, User.password == password))).scalar()
session.close()
return ret
def validate_username(self, field):
if len(field.data) < 6:
field.errors.append('Логин должен быть не менее 6 символов')
if Session.query(User).filter_by(username=field.data).first():
field.errors.append(
'Пользователь с таким логином уже зарегистрирован')
def get_user(self, username):
return Session.query(User).filter_by(username=username).first()
def main():
posts = Session.query(Post).order_by(Post.created)
return render_template("index.html", posts=posts)
def load_user(user_id):
return Session.query(User).filter_by(id=user_id).one_or_none()
def post_view(id):
post = Session.query(Post).filter_by(id=id).one_or_none()
if post:
return render_template("post/post.html", user=current_user, post=post)
else:
return redirect(url_for("home"))
def get_post(post_id):
session = Session()
post = session.query(Post).filter_by(id=post_id).scalar()
return post
def read_post(post_id):
post = Session.query(Post).filter_by(id=post_id).first()
return render_template("posts/read.html", post=post)
def get(self, user_id):
session = Session()
user = session.query(User).filter_by(id=user_id).first()
self.render('user/update.html', user=user)
def get(self):
username = self.get_argument('username', '')
session = Session()
user = session.query(User).filter_by(username=username).first()
self.render('user/selfupdate.html', user=user, username=username)
def profile(username):
user = Session.query(User).filter_by(username=username).first()
return render_template("users/profile.html", user=user)
def validate_username_unique(username):
if Session.query(User).filter_by(username=username).one_or_none():
raise BadRequest(f"User with username {username!r} already exists!")
def index():
posts = Session.query(Post).all()
return render_template('index.html', posts=posts)
def index():
if current_user.is_authenticated:
posts = Session.query(Post).order_by(desc(Post.id)).all()
return render_template("index.html", posts=posts, user=current_user)
return redirect(url_for("auth.login"))
def is_exists(cls, username, password):
session = Session()
ret = session.query(exists().where(and_(cls.username==username, cls.password==password))).scalar()
# ret = bool(user)/
session.close()
return ret
