def user_page(username):
if "userid" not in session:
return redirect('/login')
posts = PostsModel(db.get_connection())
users = UsersModel(db.get_connection())
subs = SubsModel(db.get_connection())
userid = users.get_by_name(username)[0]
all_posts = []
for i in posts.get_all(userid):
all_posts.append({
'pub_date':
datetime.fromtimestamp(i[5]).strftime('%d.%m.%Y %H:%M'),
'title':
i[1],
'thumb':
i[3],
'userid':
i[4],
'pid':
i[0]
})
user_data = users.get(userid)
user_info = {
'username': user_data[1],
'main_photo': user_data[4],
'subscribed': subs.check_subscribed(session['userid'], userid),
'userid': user_data[0]
}
user_data = users.get(session['userid'])
sess_info = {'username': user_data[1], 'main_photo': user_data[4]}
return render_template('project_index.html',
title='Instagram',
posts=all_posts,
user_info=user_info,
sess_info=sess_info)
def change_info():
if "userid" not in session:
return redirect('/login')
form = ChangeInfoForm()
if form.password_hash.data:
password_hash = generate_password_hash(form.password_hash.data)
else:
password_hash = ''
keys = {'user_name': form.user_name.data, 'password_hash': password_hash}
users = UsersModel(db.get_connection())
if form.validate_on_submit():
if not users.get_by_name(form.user_name.data):
if form.main_photo.data != '':
f = request.files['main_photo']
save_filename, thmb_filename = save_file(f)
users.update_user_info(session['userid'], 'main_photo',
thmb_filename)
for key in keys:
if keys[key] != '':
users.update_user_info(session['userid'], key, keys[key])
return redirect('/')
else:
form.username.errors.append(
'Пользователь с таким именем уже существует')
user_data = users.get(session['userid'])
user_info = {'username': user_data[1], 'main_photo': user_data[4]}
return render_template('project_change_info.html',
form=form,
user_info=user_info,
sess_info=user_info)
def index():
if "userid" not in session:
return redirect('/login')
posts = PostsModel(db.get_connection())
all_posts = []
for i in posts.get_all(session['userid']):
all_posts.append({
'pub_date':
datetime.fromtimestamp(i[5]).strftime('%d.%m.%Y %H:%M'),
'title':
i[1],
'thumb':
i[3],
'userid':
i[4],
'pid':
i[0]
})
users = UsersModel(db.get_connection())
subs = SubsModel(db.get_connection())
user_data = users.get(session['userid'])
user_info = {
'username': user_data[1],
'main_photo': user_data[4],
'followers_count': subs.get_followers(session['userid']),
'subscriptions_count': subs.get_subscriptions(session['userid']),
'posts_count': posts.get_count(session['userid']),
'userid': session['userid']
}
return render_template('project_index.html',
title='Instagram',
posts=all_posts,
user_info=user_info,
sess_info=user_info)
def login():
form = LoginForm()
login_error = ''
if form.validate_on_submit():
users = UsersModel(db.get_connection())
user = users.exists(form.username.data, form.password.data)
if user[0]:
session['userid'] = users.get(user[1])[0]
session['username'] = users.get(user[1])[1]
session['admin'] = users.get(user[1])[3]
session['sort'] = 0
speak('Login approved!')
return redirect('/')
else:
login_error = 'Invalid Login Cradentials.'
speak('Invalid Login Cradentials.')
return render_template('login.html',
title='My Diary',
brand="Personal Diary",
form=form,
login_error=login_error)
def upload_photo():
if "userid" not in session:
return redirect('/login')
form = UploadPhotoForm()
if form.validate_on_submit():
f = request.files['file']
save_filename, thmb_filename = save_file(f)
posts = PostsModel(db.get_connection())
posts.insert(form.descrypt.data, save_filename, thmb_filename,
session['userid'])
return redirect('/')
users = UsersModel(db.get_connection())
user_data = users.get(session['userid'])
user_info = {'username': user_data[1], 'main_photo': user_data[4]}
return render_template('project_add_photos.html',
form=form,
user_info=user_info,
sess_info=user_info)
def login():
form = LoginForm()
login_error = ''
if form.validate_on_submit():
users = UsersModel(db.get_connection())
# проверяем существует ли пользователь с таким логином и паролем
user = users.exists(form.username.data, form.password.data)
if user[0]:
# выставляем id пользователя в сессию
session['userid'] = users.get(user[1])[0]
return redirect('/')
else:
login_error = 'Неправильный логин или пароль'
return render_template('project_login.html',
title='Instagram',
form=form,
login_error=login_error,
user_info={'userid': 0})
def show_performances():
if 'username' not in session:
return redirect('/login')
performances_model = PerformancesModel(db.get_connection())
performances = performances_model.get_all()
if session['username'] == ADMIN_NAME:
admin = True
else:
admin = False
res = {}
users_model = UsersModel(db.get_connection())
for item in performances:
if str(item[0]) in users_model.get(session['user_id'])[3].split(','):
res[item[0]] = True
else:
res[item[0]] = False
return render_template('performances.html',
username=session['username'],
title='Спектакли',
performances=performances,
admin=admin,
in_favourites=res)
def show_page(self):
uic.loadUi('design/main.ui', self)
self.username.setText(self.current_user[1])
pixmap = QPixmap('design/little_insta.png')
if self.del_dialog != '':
self.del_dialog.close()
self.del_dialog = ''
if self.current_page != self.current_user:
subs = SubsModel(self.db.get_connection())
self.sub.show()
self.add_photo.hide()
self.edit_profile.hide()
if subs.check_subscribed(self.current_user[0],
self.current_page[0]):
self.sub.setText('Отписаться')
self.sub.setStyleSheet('font: 16pt "Yu Gothic UI Semilight";'
'color: rgb(0, 170, 255);'
'border-style: outset;'
'border-width: 2px;'
'border-radius: 10px;'
'border-color: rgb(0, 170, 255);'
'min-width: 3em;'
'padding: 6px;'
'background-color: rgb(255, 255, 255);')
else:
self.sub.setText('Подписаться')
self.sub.setStyleSheet('font: 16pt "Yu Gothic UI Semilight";'
'color: rgb(255, 255, 255);'
'border-style: outset;'
'border-width: 2px;'
'border-radius: 10px;'
'border-color: rgb(0, 170, 255);'
'min-width: 3em;'
'padding: 6px;'
'background-color: rgb(0, 170, 255);')
self.sub.clicked.connect(self.sub_unsub)
else:
self.sub.hide()
self.add_photo.show()
self.edit_profile.show()
self.icon.setPixmap(pixmap)
self.exit.clicked.connect(self.quit)
### данные о пользователе ###
posts = PostsModel(self.db.get_connection())
all_posts = []
for i in posts.get_all(self.current_page[0]):
all_posts.append({
'pub_date':
datetime.fromtimestamp(i[5]).strftime('%d.%m.%Y %H:%M'),
'title':
i[1],
'thumb':
i[3],
'userid':
i[4],
'pid':
i[0]
})
users = UsersModel(self.db.get_connection())
subs = SubsModel(self.db.get_connection())
user_data = users.get(self.current_page[0])
user_info = {
'username': user_data[1],
'main_photo': user_data[4],
'followers_count': subs.get_followers(self.current_page[0]),
'subscriptions_count':
subs.get_subscriptions(self.current_page[0]),
'posts_count': posts.get_count(self.current_page[0]),
'userid': self.current_page[0]
}
img_name = '/'.join(user_info['main_photo'].split(
'/')[:-1]) + '/circle_' + user_info['main_photo'].split('/')[-1]
if not os.path.exists(img_name):
img = Image.open(user_info['main_photo'])
img_circle = mask_circle_solid(img, (255, 255, 255), 0, offset=0)
img_circle.save(img_name)
pixmap = QPixmap(img_name)
pixmap_resized = pixmap.scaled(100, 100, QtCore.Qt.KeepAspectRatio)
self.main_photo.setPixmap(pixmap_resized)
self.username_2.setText(user_info['username'])
self.stats.setText(
'Публикации: {} Подписчики: {} Подписки: {}'.format(
user_info['posts_count'], user_info['followers_count'],
user_info['subscriptions_count']))
self.edit_profile.clicked.connect(self.edit)
self.add_photo.clicked.connect(self.add)
### начало таблицы с фотографиями ###
posts = PostsModel(self.db.get_connection())
all_posts = []
for i in posts.get_all(self.current_page[0]):
all_posts.append({
'pub_date':
datetime.fromtimestamp(i[5]).strftime('%d.%m.%Y %H:%M'),
'title':
i[1],
'thumb':
i[3],
'userid':
i[4],
'pid':
i[0]
})
self.table.setColumnCount(3)
self.row_count = user_info['posts_count'] // 3 if user_info[
'posts_count'] % 3 == 0 else user_info['posts_count'] // 3 + 1
self.table.setRowCount(self.row_count)
for i in range(0, self.row_count):
for j in range(0, 3):
if i * 3 + j < len(all_posts):
image = ImageWidget(all_posts[i * 3 + j]['thumb'], self)
self.table.setCellWidget(i, j, image)
self.table.cellClicked[int, int].connect(self.show_post)
### поиск пользователя ###
self.find_user.clicked.connect(self.search_user)
self.username.clicked.connect(self.return_home)
def show_post(self, *params):
if len(params) > 1:
self.r = params[0]
self.c = params[1]
uic.loadUi('design/show_post.ui', self)
if self.del_dialog != '':
self.del_dialog.close()
self.del_dialog = ''
if self.current_page != self.current_user:
self.settings_combo.hide()
self.send.show()
self.save_desc.hide()
self.username.setText(self.current_page[1])
img_name = '/'.join(self.current_page[4].split(
'/')[:-1]) + '/circle_' + self.current_page[4].split('/')[-1]
pixmap = QPixmap(img_name)
pixmap_resized = pixmap.scaled(71, 61, QtCore.Qt.KeepAspectRatio)
self.main_photo.setPixmap(pixmap_resized)
self.arrow.clicked.connect(self.show_page)
posts = PostsModel(self.db.get_connection())
all_posts = []
for i in posts.get_all(self.current_page[0]):
all_posts.append({
'pub_date':
datetime.fromtimestamp(i[5]).strftime('%d.%m.%Y %H:%M'),
'title':
i[1],
'thumb':
i[3],
'userid':
i[4],
'pid':
i[0]
})
if self.r * 3 + self.c > len(all_posts) - 1:
self.show_page()
return
self.cur_post = posts.get(all_posts[self.r * 3 + self.c]['pid'])
pixmap = QPixmap(self.cur_post[2])
pixmap_resized = pixmap.scaled(781, 401, QtCore.Qt.KeepAspectRatio)
r_width = pixmap_resized.width()
self.photo.move(957 // 2 - r_width // 2, 100)
self.photo.setPixmap(pixmap_resized)
likes = LikesModel(self.db.get_connection())
count = likes.get_count(self.cur_post[0])
self.likes.setText(str(count))
if likes.get_your(self.cur_post[0], self.current_user[0]):
self.like.setIcon(QIcon('design/heart_red.png'))
self.like.setIconSize(QtCore.QSize(51, 51))
self.like.clicked.connect(self.del_like)
else:
self.like.setIcon(QIcon('design/heart_white.png'))
self.like.setIconSize(QtCore.QSize(51, 51))
self.like.clicked.connect(self.add_like)
comments_ = CommentsModel(self.db.get_connection())
users = UsersModel(self.db.get_connection())
comments = comments_.get(self.cur_post[0])
comment_count = len(comments)
self.comments.setRowCount(comment_count)
for i in range(0, comment_count):
user = users.get(comments[i][2])[1]
item = QTableWidgetItem(user)
item.setFont(QFont('Yu Gothic UI Semibold', 16))
item.setTextAlignment(QtCore.Qt.AlignRight)
self.comments.setItem(i, 0, item)
self.comments.setItem(i, 1, QTableWidgetItem(comments[i][4]))
time = QLabel(time_converter(comments[i][5]))
time.setStyleSheet(
'color: rgb(208, 208, 208); font: Yu Gothic UI Semibold; font-size: 16px;'
)
self.comments.setCellWidget(i, 2, time)
if comments[i][2] == self.current_user[0] and comments[i][0] != -1:
image = QLabel('картинка')
pixmap = QPixmap('design/can.png')
pixmap_resized = pixmap.scaled(15, 15,
QtCore.Qt.KeepAspectRatio)
image.setPixmap(pixmap_resized)
self.comments.setCellWidget(i, 3, image)
self.comments.resizeColumnsToContents()
self.comments.cellClicked[int, int].connect(self.del_comment)
self.send.clicked.connect(self.add_comment)
self.settings_combo.setItemIcon(0, QIcon('design/tools.png'))
self.settings_combo.activated[str].connect(self.settings)
