Update app.py

app.py

@@ -10,6 +10,7 @@ from telegram.ext import (Updater,
 from telegram import InlineKeyboardButton, InlineKeyboardMarkup, ForceReply, ReplyKeyboardMarkup
 import requests
 import hashlib
+import sqlite3
 import re
 from bs4 import BeautifulSoup
 
@@ -22,6 +23,12 @@ EXPECT_STATUS, EXPECT_USERNAME, EXPECT_PASSWORD, EXPECT_SUBMIT, EXPECT_BACK, EXP
 
 def start(update: Update, context: CallbackContext):
 	chat_id = update.effective_chat.id
+	user_data = check_existance(chatID=chat_id)
+	if not user_data:
+		create_user(chatID=chat_id, username='', password='')
+		user_data = {'id': 0, 'chatID': chat_id, 'username': None, 'password': None}
+	else:
+		user_data = dict(zip(['id', 'chatID', 'username', 'password']))
 	text = ('به ربات پرتال دانشجویی خوش آمدید 🌹')
 	update.message.reply_text(text, reply_markup=main_keyboard)
 
@@ -69,9 +76,9 @@ def button_click_handler(update: Update, context: CallbackContext):
 
 
 def hash_password(text):
-    password = hashlib.md5()
-    password.update(text.encode('utf-8'))
-    return password.hexdigest()
+	password = hashlib.md5()
+	password.update(text.encode('utf-8'))
+	return password.hexdigest()
 
 
 def login_portal(update: Update, context: CallbackContext):
@@ -115,29 +122,61 @@ def set_username(update: Update, context: CallbackContext):
 
 
 def cancel(update: Update, context: CallbackContext):
-    update.message.reply_text('منوی اصلی', reply_markup=main_keyboard)
-    return ConversationHandler.END
-
+	update.message.reply_text('منوی اصلی', reply_markup=main_keyboard)
+	return ConversationHandler.END
 
 #############################################
 
-def captcha_url():
-	web_url = 'https://pooya.um.ac.ir/gateway/PuyaAuthenticate.php'
-	req_url = 'https://pooya.um.ac.ir/gateway/UserInterim.php'
-	headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/103.0.0.0 Safari/537.36'}
-	session = requests.session()
-	resp = session.get(web_url, headers=headers)
-	soup = BeautifulSoup(resp.content, 'html.parser')
-	img_src = soup.find('img', id='secimg')['src']
-	capcha_src = 'https://pooya.um.ac.ir' + img_src[2:]
-	return capcha_src
+def create_database(update: Update, context: CallbackContext):
+	chat_id = update.effective_chat.id
+	if chat_id == admin:
+		try:
+			con.execute('CREATE TABLE USER (id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT, chatID TEXT, username TEXT, password TEXT)')
+			update.message.reply_text('دیتابیس با موفقیت ساخته شد', reply_markup=main_keyboard)
+		except:
+			update.message.reply_text('دیتابیس قبلا ایجاد شده است', reply_markup=main_keyboard)
+
+
+def show_table(update: Update, context: CallbackContext):
+	chat_id = update.effective_chat.id
+	if chat_id == admin:
+		records = con.execute('SELECT * FROM USER')
+		table = [', '.join(map(str,user)) for user in records]
+		text = '\n'.join(table)
+		update.message.reply_text(text, reply_markup=main_keyboard)
+
+
+def create_user(chatID, username, password):
+	con.execute(f'INSERT INTO USER (chatID, username, password) values({chatID}, {username}, {password})')
+
+
+def check_existance(chatID, verbose=True):
+	records = con.execute('SELECT * FROM USER')
+	if verbose:
+		for user in records:
+			if user[1] == chatID:
+				return user
+	return None
+
+			
+def update_user(chatID, username, password):
+	con.execute(f'UPDATE USER (username, password) values({username}, {password}) WHERE chatID = {chatID}')
+
+#############################################
 
 if __name__ == '__main__':
+	# Database
+	db_url = 'my2-test.db'
+	db_con = sqlite3.connect(db_url)
+	con = db_con.cursor()
+	# Session
 	session = requests.session()
 	headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/103.0.0.0 Safari/537.36'}
-	pers = PicklePersistence(filename='mybot')
-	updater = Updater(token=bot_token, persistence=pers)
+	# Bot details
+	updater = Updater(token=bot_token)
 	updater.dispatcher.add_handler(CommandHandler('start', start))
+	updater.dispatcher.add_handler(CommandHandler('db', create_database))
+	updater.dispatcher.add_handler(CommandHandler('all', show_table))
 	updater.dispatcher.add_handler(ConversationHandler(
 		entry_points=[MessageHandler(Filters.regex('ورود به پرتال دانشجویی'), set_info_handler)],
 		states={