def wrapper(*args, **kwargs): user_info = database.check_user(kwargs['event'].data["from"]["userId"]) if user_info["not_exist"]: database.add_user(kwargs['event'].data["from"]["userId"]) user_info = database.check_user( kwargs['event'].data["from"]["userId"]) func(*args, **kwargs)
def wrapper(*args, **kwargs): user_info = database.check_user(kwargs['event'].data["from"]["userId"]) if user_info["not_exist"]: database.add_user(kwargs['event'].data["from"]["userId"], kwargs['event'].data["from"]["firstName"], kwargs['event'].data["from"].get('lastName', '')) user_info = database.check_user( kwargs['event'].data["from"]["userId"]) func(*args, **kwargs)
def auth_signup(): post_data = request.get_json() from database import check_user user = check_user(post_data['username'], post_data['password']) print(user) if user is not None and (user['email'] is None or user['email'] == ''): from database import update_user if update_user(post_data['username'], post_data['email'], post_data['fullname']): token = jwt.encode({ 'time': datetime.now().strftime('%Y-%m-%d %H:%M:%S'), 'username': post_data['username'] }, JWT_SECRET, algorithm=JWT_ALGORITHM).decode('utf-8') return { 'success': True, 'token': token } else: return { 'success': False, 'reason': 'creation_failed' } return { 'success': False, 'reason': 'no_account' if user is None else 'already_created' }
def check_user(self, username, password): if db.check_user(username, password): print("welcome ", username) self.show_frame(MainMenu) else: print("who are you??")
def process(message_data): user_db = database.connect_db() type = message_data.get('post_type') message_type = message_data.get('message_type') if type == 'message' and message_type == 'private': uid = message_data.get('user_id') message = message_data.get('message') exist = database.check_user(uid, user_db) matched = re.search(pattern=r"[(.*?)]", string=message, flags=re.M | re.I) if exist == 0: if matched: database.user_register(uid, matched.group(1), user_db) return 0 else: send._pri_m('数据库中并没有查询到您的数据呢QAQ\n来注册一个吧', uid) send._pri_m('注册格式为: [您喜欢的名称] (保留方括号噢@A@)', uid) send._pri_m('请尽量使用英文字母噢(*^_^*)', uid) return 0 else: reply = response.reply(uid, type, message) data = reply.get_response() if data != 'ERROR': send._pri_m(data, uid) return 0 elif type == 'message' and message_type == 'group': uid = message_data.get('user_id') gid = message_data.get('group_id') message = message_data.get('message') self_id = message_data.get('self_id') if not database.check_user(uid, user_db): stri = '数据库中并没有查询到您的数据呢QAQ\n来注册一个吧\n向机器人私聊发送消息完成注册' send._group_m(stri, gid, uid) return 0 reply = response.group_reply(str(self_id), uid, type, message) data = reply.get_response() print data if data != 'ERROR': send._group_m(data, gid, uid) return 0 elif type == 'requests': uid = message_data.get('user_id') flag = message_data.get('flag') send._acc_request(flag) return 0
def check_user(): username = login_tbox1Text.get() password = login_tbox2Text.get() if db.check_user(username, password): print("welcome ", username) else: print("who are you??")
def login(): POST_USERNAME = str(request.form['username']) POST_PASSWORD = str(request.form['password']) res = check_user(POST_USERNAME, POST_PASSWORD) if res: session['logged_in'] = True session['team_id'] = res.id return redirect('/')
def f_login(): if request.method == 'POST': user = request.form['username'] passw = request.form['password'] check = request.form.getlist('check') if check_user(user, passw): if check: session['user'] = user return render_template("ad-search.html", Username=user) return render_template("main.html", items=get_4())
def main(): options = int(input("1: Add User" "\n2: Delete User" "\n3: Check User" "\n4: Display Db" "\n5: Create Db\n")) if options == 1: try: database.new_user() except Exception as error: print("ERROR:", error) finally: print("\n") main() if options == 2: try: database.delete_user() except Exception as error: print("ERROR:", error) finally: print("\n") main() if options == 3: try: database.check_user() except Exception as error: print("ERROR:", error) finally: print("\n") main() if options == 4: database.display_db() print("\n") main() if options == 5: try: database.create_db() print("Database created\n") except Exception as error: print("Error", error) finally: main()
def login(): if request.method == 'POST': email = request.form['email'] password = request.form['password'] if db.check_user(email, password): print('logged in') return render_template('hello.html', name=email) else: return "password or username is incorrect" return render_template('login.html')
def access_db(): """Renders 'user.html' page and checks for passed username inside the database. Gets user's data tuple only when flash messages are not displayed.""" form = LoginForm() global session_user if form.validate_on_submit(): if not signing_up: # As signing in... if database.check_user(form.username.data) is not None: flash('Username taken!') return redirect(url_for('access_db')) else: database.create_user(form.username.data, form.userkey.data) session_user = database.check_user(form.username.data) return redirect(url_for('user_choice')) else: if database.check_user(form.username.data) is None: flash('Cannot find username in database!') else: session_user = database.check_user(form.username.data) return redirect(url_for('user_choice')) return render_template('user.html', form=form)
def login(): if request.method == 'POST': email = request.form['email'] password = request.form['password'] if db.check_user(email, password): print('logged in') session['email'] = email return redirect(url_for('index')) else: return "password or username is incorrect" # Get request return render_template('login.html')
def login(): message = '' if request.method == 'POST': email = request.form.get('email') password_input = request.form.get('password') user = get_user(email) print(user) if check_user(email, password_input): login_user(user) return redirect(url_for('home')) else: message = 'Failed to login!' return render_template('login.html', message=message)
def start(update, context): tel_id = update.message.chat.id if not db.check_user(tel_id): keys = [[ InlineKeyboardButton('آقا', callback_data='male'), InlineKeyboardButton('خانم', callback_data='female') ]] update.message.reply_text( 'سلام، به ربات دوستیاب خوش آمدید \n\n' 'برای شروع ابتدا جنسیت خود را انتخاب کنید', reply_markup=InlineKeyboardMarkup(keys, one_time_keyboard=True)) else: gender = db.get_gender(tel_id)[0] if gender == 'male': choices_male(update, context, False) elif gender == 'female': choices_female(update, context, False)
def auth(request): body = json.loads(request.body) if body["type"] == "isLog": for this_session in request.session.keys(): #If session exists yet return HttpResponse(json.dumps({"isLog": True })) #Return true if session exists return HttpResponse(json.dumps( {"isLog": False})) #Return false if session doesn't exist elif body["type"] == "login": if database.check_user( body["login"], body["password"] ) == True: #Checking if login and password correct request.session[body["login"]] = body[ "password"] #Start new session return HttpResponse( json.dumps({"redirect_url": url + "?profile=" + body["login"]}) ) #Return redirection link to profile page after success login else: return HttpResponse( json.dumps({'error': 'Incorrect login or password' })) #Error if login or password incorrect elif body["type"] == "logout": for this_session in request.session.keys(): #Get session name del request.session[this_session] #Delete session with this name break #Stop loop return HttpResponse(json.dumps({"redirect_url": url + "auth" })) #Return redirection link to auth elif body["type"] == "registration": if database.check_login( body["login"]) == True: #Checking login uniqueness return HttpResponse(json.dumps({'error': 'User already exist' })) #Error if this login exist yet database.add_user(body["name"], body["login"], body["password"]) #Create new user request.session[body["login"]] = body["password"] #Start new session return HttpResponse( json.dumps({ "redirect_url": url + "?profile=" + body["login"] })) #Return redirection link to profile page after registration
def auth_login(): post_data = request.get_json() from database import check_user from database import clear_user_token user = check_user(post_data['username'], post_data['password']) if user is not None and (user['email'] is not None and user['email'] != ''): clear_user_token(post_data['username']) token = jwt.encode({ 'time': datetime.now().strftime('%Y-%m-%d %H:%M:%S'), 'username': post_data['username'] }, JWT_SECRET, algorithm=JWT_ALGORITHM).decode('utf-8') return { 'success': True, 'token': token, 'fullName': user['fullName'], 'email': user['email'], 'webAlerts': user['webAlerts'], 'emailAlerts': user['emailAlerts'] } return { 'success': False, 'reason': 'no_account' if user is None else 'signup_required' }
def check_new_msg(self): url = self.url + "getUpdates?offset={}".format(self.offset) content = self.get_url(url) if content == None: return try: update_id = content['update_id'] txt = content['message']['text'] user_id = content['message']['from']['id'] self.update_offset(update_id) if int(check_user(user_id)) == 1: if get_state(user_id,"on_register")[0][0] == 1: txt = txt.strip(" ") try: text = txt.split(",") res = change_parameters(user_id,text[0],text[1],text[2],text[3], text[4]) except Exception as e: print(e) res = 0 if res ==1: self.send_message(user_id, "Valores registrados con exito!") reset_field(user_id,"ack_eth") reset_field(user_id,"ack_btc") else: self.send_message(user_id, "Hubo un problema registrando sus valores. Intentelo nuevamente") reset_field(user_id, "on_register") else: print(txt) if txt == '1': history_file = open("./history.txt") last_prices = history_file.readlines()[-1] history_file.close() last_prices = last_prices.strip("\n") last_prices = last_prices.split(",") self.send_message(user_id,"Los ultimos precios registrados son:\nFECHA: "+last_prices[0]+"\nBTC = "+last_prices[1]+" USD"+"\nETH = "+last_prices[2]+" USD") elif txt == '2': generate_plot('BTC', get_state(user_id, 'hours')[0][0]) generate_plot('ETH', get_state(user_id, 'hours')[0][0]) self.send_img(user_id,"./btc.png") self.send_img(user_id,"./eth.png") self.send_img(user_id,"./btc_candlestick.png") self.send_img(user_id,"./eth_candlestick.png") elif txt == '3': self.send_message(user_id,"Para registrarse correctamente, escriba los limites superiores e inferiores de su alarma separados por coma. Si no desea tener en cuenta un parametro escriba -1 en su valor. Por ejemplo: btc_sup,btc_inf,eth_sup,eth_inf, hours.\n") set_field(user_id, "on_register") elif txt == '4': th_sup_btc = get_state(user_id,'th_sup_btc')[0][0] th_inf_btc = get_state(user_id,'th_inf_btc')[0][0] th_sup_eth = get_state(user_id,'th_sup_eth')[0][0] th_inf_eth = get_state(user_id,'th_inf_eth')[0][0] hours = str(get_state(user_id,'hours')[0][0]) self.send_message(user_id,"Tus parametros actuales son:\nLIMITE SUPERIOR BTC = " +th_sup_btc+"\nLIMITE INFERIOR BTC = "+th_inf_btc+"\nLIMITE SUPERIOR ETH = "+th_sup_eth+"\nLIMITE INFERIOR ETH = "+th_inf_eth+"\nVENTANA DE TIEMPO PARA GRAFICOS = "+hours+"h") elif txt == 'btc_ok': set_field(user_id , "ack_btc") self.send_message(user_id, " Alarma de BTC reconocida!") print("BTC ACK = 1") elif txt == 'eth_ok': set_field(user_id , "ack_eth") self.send_message(user_id, " Alarma de ETH reconocida!") print("ETH ACK = 1") else: self.display_options(user_id) else: self.send_message(user_id,"Para registrarse correctamente, escriba los limites superiores e inferiores de su alarma separados por coma. Si no desea tener en cuenta un parametro escriba -1 en su valor. Por ejemplo: btc_sup,btc_inf,eth_sup,eth_inf, hours.\n") print(add_user(user_id)) set_field(user_id, "on_register") except Exception as e: self.update_offset(update_id) print(e) pass
def login(): global who global currentUserID all_disabled() if len(entry_userId.get()) != 0 and len(entry_pass.get()) != 0: userID = database.check_user(entry_userId.get(), entry_pass.get()) if userID: if userID == "0": who = 0 button_take.configure(state='normal') button_give.configure(state='normal') elif userID == "1": who = 1 button_middle.configure(state='normal') button_add.configure(state='normal') button_del.configure(state='normal') button_take.configure(state='normal') button_give.configure(state='normal') button_frequency.configure(state='normal') button_onHand.configure(state='normal') elif userID == "2": who = 2 button_middle.configure(state='normal') button_add.configure(state='normal') button_del.configure(state='normal') button_take.configure(state='normal') button_give.configure(state='normal') button_plusOne.configure(state='normal') button_plusTwo.configure(state='normal') button_plusFive.configure(state='normal') button_plusTen.configure(state='normal') button_plusFT.configure(state='normal') button_plusTwenty.configure(state='normal') button_frequency.configure(state='normal') button_onHand.configure(state='normal') elif userID == "5": messagebox.showerror('error', 'Неверный пароль') return else: messagebox.showerror('error', 'Пользователь не найден') else: messagebox.showerror('error', 'Заполните необходимые поля') return var1.set(0) var2.set(0) var3.set(0) if len(entry_userId.get()) != 0: currentUserID = entry_userId.get() fill_on_hand_table() entry_userId.delete(0, 'end') entry_pass.delete(0, 'end') button_sortID.configure(state='normal') button_sortID2.configure(state='normal') button_sortName.configure(state='normal') button_sortName2.configure(state='normal') button_sortAuthor.configure(state='normal') button_sortAuthor2.configure(state='normal') button_sortYear.configure(state='normal') button_sortYear2.configure(state='normal') button_sortCount.configure(state='normal') button_sortCount2.configure(state='normal') button_exit.configure(state='normal') button_enter.configure(state='disabled') button_reg.configure(state='disabled')
def pword(db, tok): key = itsdangerous.URLSafeSerializer(config.cred['secretKeys']) out = key.loads(tok) res = database.check_user(db, out[0], out[1]) if res: return template('pwordReset', user=tok, pwordError=False)
def robot(message): if message.text == "/start": if message.from_user.id in config.ADMIN_ID: markup = types.InlineKeyboardMarkup() a = types.InlineKeyboardButton("Iran Cyber Security Group", url="https://iran-cyber.net") b = types.InlineKeyboardButton("Safe Data Hosting Co", url="https://safe-data.ir") markup.add(a) markup.add(b) bot.send_message(message.chat.id, "Yay, Admin 😎🌹\nSend me your Link", reply_markup=markup) else: if database.check_user(message.from_user.id) == "New_User": markup = types.InlineKeyboardMarkup() a = types.InlineKeyboardButton("En", callback_data="En") b = types.InlineKeyboardButton("Fa", callback_data="Fa") markup.add(a, b) bot.send_message( message.chat.id, "سلام [{}](tg://user?id={})\nبه ربات کوتاه کننده لینک ایران سایبر خوش آمدید\nزبان خود را انتخاب کنید\n➖➖➖➖➖➖\nHi [{}](tg://user?id={})\nWelcome to IRan Cyber Shortlink Bot\nSelect Your Language" .format(message.from_user.first_name, message.from_user.id, message.from_user.first_name, message.from_user.id), reply_markup=markup, parse_mode="Markdown") for x in config.ADMIN_ID: bot.send_message( x, "New user -> [{}](tg://user?id={})".format( message.from_user.first_name, message.from_user.id), parse_mode="markdown") else: if database.check_lang(message.from_user.id) == "Fa": markup = types.InlineKeyboardMarkup() change_lang = types.InlineKeyboardButton( "تغییر زبان", callback_data="change_lang") a = types.InlineKeyboardButton( "Iran Cyber Security Group", url="https://iran-cyber.net") b = types.InlineKeyboardButton("Safe Data Hosting Co", url="https://safe-data.ir") markup.add(change_lang) markup.add(a) markup.add(b) bot.send_message( message.chat.id, "سلام [{}](tg://user?id={})\nبه ربات کوتاه کننده لینک ایران سایبر خوش آمدید\nلینک خود را برای کوتاه شدن ارسال کنید" .format(message.from_user.first_name, message.from_user.id), reply_markup=markup, parse_mode="Markdown") else: markup = types.InlineKeyboardMarkup() change_lang = types.InlineKeyboardButton( "change language", callback_data="change_lang") a = types.InlineKeyboardButton( "Iran Cyber Security Group", url="https://iran-cyber.net") b = types.InlineKeyboardButton("Safe Data Hosting Co", url="https://safe-data.ir") markup.add(change_lang) markup.add(a) markup.add(b) bot.send_message( message.chat.id, "Hi [{}](tg://user?id={})\nWelcome to IRan Cyber Shortlink Bot\nEnter Your Link" .format(message.from_user.first_name, message.from_user.id), reply_markup=markup, parse_mode="Markdown") else: if message.text.startswith("http://") or message.text.startswith( "https://"): markup = types.InlineKeyboardMarkup() a = types.InlineKeyboardButton("Iran Cyber Security Group", url="https://iran-cyber.net") b = types.InlineKeyboardButton("Safe Data Hosting Co", url="https://safe-data.ir") markup.add(a) markup.add(b) if message.from_user.id in config.ADMIN_ID: bot.send_message(message.chat.id, "Shorten Link : \n{}".format( short_link(message.text)), reply_markup=markup) else: if database.check_lang(message.from_user.id) == "Fa": bot.send_message(message.chat.id, "لینک کوتاه شده ی شما \n{}".format( short_link(message.text)), reply_markup=markup) else: bot.send_message(message.chat.id, "Shorten Link : \n{}".format( short_link(message.text)), reply_markup=markup)