Update app.py

app.py

@@ -10,28 +10,111 @@ from telegram.ext import (Updater,
 from telegram import InlineKeyboardButton, InlineKeyboardMarkup, ForceReply, ReplyKeyboardMarkup
 import requests
 import hashlib
-from random import randint
+import re
 from bs4 import BeautifulSoup
-import urllib.request
 
 bot_token = '5415345086:AAEpiZypEMxgEs5zMcyOBAx_CpT9iLcnE8s'
 admin = 37087739
 
+#############################################
+
+EXPECT_STATUS, EXPECT_USERNAME, EXPECT_PASSWORD, EXPECT_SUBMIT, EXPECT_BACK, EXPECT_BUTTON_CLICK  = range(6)
+
 def start(update: Update, context: CallbackContext):
-	text = ('Hello!')
-	update.message.reply_text(text)
+	chat_id = update.effective_chat.id
+	text = ('Hello dear user! 🌹\n'
+			'Please enter the username or instagram url\n\n'
+			'Powered by @eskandani')
+	update.message.reply_text(text, reply_markup=main_keyboard)
+
+
+def set_info_handler(update: Update, context: CallbackContext):
+	username = context.user_data.get('username', 'تنظیم نشده است')
+	password = context.user_data.get('password', 'تنظیم نشده است')
+	text = (f'نام کاربری: {username}\n'
+			f'رمز عبور: {password}\n\n'
+			'لطفا یک گزینه را انتخاب نمایید')
+	button = [  [InlineKeyboardButton('نام کاربری', callback_data='username'),
+				InlineKeyboardButton('رمز عبور', callback_data='password')],
+				[InlineKeyboardButton('ورود و دریافت نمرات', callback_data='login_portal')],
+				[InlineKeyboardButton('بازگشت', callback_data='cancel')],
+			]
+	inline_keyboard = InlineKeyboardMarkup(button)
+	update.message.reply_text(text, reply_markup=inline_keyboard)
+	return EXPECT_BUTTON_CLICK
+
+def button_click_handler(update: Update, context: CallbackContext):
+	query = update.callback_query
+	query.answer()
+	query.delete_message()
+	if query.data == 'login_portal':
+		web_url = 'https://pooya.um.ac.ir/gateway/PuyaAuthenticate.php'
+		resp = session.get(web_url, headers=headers)
+		soup = BeautifulSoup(resp.content, 'html.parser')
+		img_src = soup.find('img', id='secimg')['src']
+		filename = 'C:/Users/Home/OneDrive/Desktop/captcha.png'
+		capcha_src = 'https://pooya.um.ac.ir' + img_src[2:]
+		context.user_data['captcha'] = capcha_src
+		context.bot.send_message(chat_id, text=capcha_src)
+		context.bot.send_message(chat_id, text='کد کپچا را وارد کنید', reply_markup=ForceReply())
+		return EXPECT_SUBMIT
+	elif query.data == 'username':
+		context.bot.send_message(chat_id, text='نام کاربری خود را وارد کنید', reply_markup=ForceReply())
+		return EXPECT_USERNAME
+	elif query.data == 'password':
+		context.bot.send_message(chat_id, text='رمز عبور خود را وارد کنید', reply_markup=ForceReply())
+		return EXPECT_PASSWORD
+	elif query.data == 'cancel':
+		query.answer('Opeartion canceled!')
+		return ConversationHandler.END
+
+
+def hash_password(text):
+    password = hashlib.md5()
+    password.update(text.encode('utf-8'))
+    return password.hexdigest()
+
 
-def send_captcha(update: Update, context: CallbackContext):
-	''' Entry point of conversation  this gives  buttons to user'''
+def login_portal(update: Update, context: CallbackContext):
 	chat_id = update.effective_chat.id
-	context.bot.send_message(chat_id, text='Salam')
-	url = captcha_url()
-	context.bot.send_message(chat_id, text=url)
-	context.bot.send_photo(chat_id, photo=url)
-	urllib.request.urlretrieve(url, 'captcha.png')
-	context.bot.send_message(chat_id, text='Downloaded')
-	
-	context.bot.send_photo(chat_id, photo=open('captcha.png', 'rb'))
+	captcha = update.message.text
+	USER = context.user_data['username']
+	PASS = context.user_data['password']
+	capcha_src = context.user_data['captcha']
+	login_data = {  'UserPassword': hash_password(PASS),
+					'pswdStatus': '',
+					'UserID': str(USER),
+					'DummyVar': '',
+					'rand2': capcha_src[capcha_src.find('rand2=')+6:],
+					'mysecpngco': captcha_val
+				}
+	req_url = 'https://pooya.um.ac.ir/gateway/UserInterim.php'
+	resp = session.post(req_url, data=login_data) 
+	if resp.status_code == 200:
+		grade_list = session.get('https://pooya.um.ac.ir/educ/educfac/stuShowEducationalLogFromGradeList.php', headers=headers)
+		soup = BeautifulSoup(grade_list.content, 'html.parser')
+		data = [item.text.splitlines() for item in soup.find_all('tr', bgcolor="#E8E8FF")]
+		mean = [item.text.splitlines() for item in soup.find_all('tr', bgcolor="#C4C4FF")][1][3]
+		mark = '\n'.join([f'{i+1}) {row[2]}, {row[5]} ({row[8]})' for i, row in enumerate(data)])
+		text = f'{mark}\nمعدل: {mean}'
+		context.bot.send_message(chat_id, text)
+	else:
+		print('Login faild!')
+	return ConversationHandler.END
+
+
+def set_password(update: Update, context: CallbackContext):
+	context.user_data['password'] = update.message.text
+	update.message.reply_text('رمز عبور با موفقیت ثبت شد', reply_markup=main_keyboard)
+	return ConversationHandler.END
+
+
+def set_username(update: Update, context: CallbackContext):
+	context.user_data['username'] = update.message.text
+	update.message.reply_text('نام کاربری با موفقیت ثبت شد', reply_markup=main_keyboard)
+	return ConversationHandler.END
+
+#############################################
 
 def captcha_url():
 	web_url = 'https://pooya.um.ac.ir/gateway/PuyaAuthenticate.php'
@@ -45,8 +128,24 @@ def captcha_url():
 	return capcha_src
 
 if __name__ == '__main__':
-	updater = Updater(token=bot_token)
+	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)
 	updater.dispatcher.add_handler(CommandHandler('start', start))
-	updater.dispatcher.add_handler(CommandHandler('img', send_captcha))
+	updater.dispatcher.add_handler(ConversationHandler(
+		entry_points=[MessageHandler(Filters.regex('ورود به پرتال دانشجویی'), set_info_handler)],
+		states={
+			EXPECT_USERNAME: [MessageHandler(Filters.text, set_username)],
+			EXPECT_PASSWORD: [MessageHandler(Filters.text, set_password)],
+			EXPECT_SUBMIT: [MessageHandler(Filters.text, login_portal)],
+			EXPECT_BUTTON_CLICK: [CallbackQueryHandler(button_click_handler)],
+			EXPECT_BACK : [CallbackQueryHandler('cancel', start)]
+		},
+		fallbacks=[MessageHandler(Filters.regex('cancel'), cancel)],
+	))
+	buttons = [['ورود به پرتال دانشجویی']]
+	main_buttons = [row[::-1] for row in buttons]
+	main_keyboard = ReplyKeyboardMarkup(main_buttons, one_time_keyboard=True, resize_keyboard=True)
 	updater.start_polling()
 	print('Bot started ..........')