def add_user(bot, update): root_from_set = config['TELEGRAM']['Root_users'] root_from_set = root_from_set.split(',') is_root = False for id in root_from_set: if str(id) == str(update.message.chat_id): is_root = True if not is_root: return False text = update.message.text text = text.replace('/addUser ', '') text = text.split(' ') adding_chat_it = text[0] try: adding_user_name = text[1] except IndexError: adding_user_name = '' session = Session() q = session.query(TrustedChats).filter( TrustedChats.chat_id == str(adding_chat_it)) if q.count(): bot.send_message(chat_id=update.message.chat_id, text="This user already in base.") bot.send_message(chat_id=update.message.chat_id, text="Bitch") else: myobject = TrustedChats(chat_id=str(adding_chat_it), name=adding_user_name) session.add(myobject) session.commit() bot.send_message(chat_id=update.message.chat_id, text="User was added.") session.close()
def addUser(username, refresh_token, playlist_id_short=None, playlist_id_medium=None, playlist_id_long=None): session = Session() id_exists = session.query(User.id).filter_by(username=username).scalar() # new user if id_exists == None: user = User(username=username, refresh_token=refresh_token, playlist_id_short=playlist_id_short, playlist_id_medium=playlist_id_medium, playlist_id_long=playlist_id_long) session.add(user) logging.info('New auto user: '******'Auto user updated: ' + user.username) # only update playlist IDs that are new if playlist_id_short != None: user.playlist_id_short = playlist_id_short if playlist_id_medium != None: user.playlist_id_medium = playlist_id_medium if playlist_id_long != None: user.playlist_id_long = playlist_id_long session.commit() session.close()
def add(id, name): session = Session() q = session.query(TrustedChats).filter(TrustedChats.chat_id == str(id)) if q.count(): click.secho(click.style('ChatId must be unique.', fg='red', blink=True), err=True) click.secho("Use \"list\" to see all trusted chat ids.") else: myobject = TrustedChats(chat_id=str(id), name=name) session.add(myobject) session.commit() click.echo('ChatId {} was added.'.format(id)) session.close()
"email":"*****@*****.**" }, { "username":"******", "email":"*****@*****.**" }, { "username":"******", "email":"*****@*****.**" }, ] local_session=Session(bind=engine) new_user=User(username="******",email="*****@*****.**") local_session.add(new_user) local_session.commit() # for u in users: # new_user=User(username=u["username"],email=u["email"]) # local_session.add(new_user) # local_session.commit() # print(f"Added {u['username']}")
furious_7.actors = [dwayne_johnson] pain_and_gain.actors = [dwayne_johnson, mark_wahlberg] # 7 - add contact details to actors matt_contact = ContactDetails("415 555 2671", "Burbank, CA", matt_damon) dwayne_contact = ContactDetails("423 555 5623", "Glendale, CA", dwayne_johnson) dwayne_contact_2 = ContactDetails("421 444 2323", "West Hollywood, CA", dwayne_johnson) mark_contact = ContactDetails("421 333 9428", "Glendale, CA", mark_wahlberg) # 8 - create stuntmen matt_stuntman = Stuntman("John Doe", True, matt_damon) dwayne_stuntman = Stuntman("John Roe", True, dwayne_johnson) mark_stuntman = Stuntman("Richard Roe", True, mark_wahlberg) # 9 - persists data session.add(bourne_identity) session.add(furious_7) session.add(pain_and_gain) session.add(matt_contact) session.add(dwayne_contact) session.add(dwayne_contact_2) session.add(mark_contact) session.add(matt_stuntman) session.add(dwayne_stuntman) session.add(mark_stuntman) # 10 - commit and close session session.commit() session.close()