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 updatePlaylists():
session = Session()
# attempt to update each user's playlists
for user in session.query(User):
is_active = False
# authorize the application with Spotify API
payload = refreshToken(user.refresh_token)
# if user account has been removed or authorization revoked, user is deleted
if payload == None:
session.delete(user)
else:
access_token = payload[0]
playlist = user.playlist_id_short
if playlist != None:
# if the playlist has not been deleted
if (dbClearPlaylist(access_token, playlist) != None):
uri_list = dbGetTopTracksURI(access_token, 'short_term',
50)
dbAddTracksPlaylist(access_token, playlist, uri_list)
is_active = True
else:
user.playlist_id_short = None
playlist = user.playlist_id_medium
if playlist != None:
if (dbClearPlaylist(access_token, playlist) != None):
uri_list = dbGetTopTracksURI(access_token, 'medium_term',
50)
dbAddTracksPlaylist(access_token, playlist, uri_list)
is_active = True
else:
user.playlist_id_medium = None
playlist = user.playlist_id_long
if playlist != None:
if (dbClearPlaylist(access_token, playlist) != None):
uri_list = dbGetTopTracksURI(access_token, 'long_term', 50)
dbAddTracksPlaylist(access_token, playlist, uri_list)
is_active = True
else:
user.playlist_id_long = None
# if no playlists could be updated, then remove user
if not is_active:
session.delete(user)
session.commit()
session.close()
logging.info('Updated TopTracks Playlists')
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()
#!/usr/bin/env python from main import Base, Session, User # 创建表结构和数据 Base.metadata.create_all() #中是我西iugai的的将阿黄飞鸿第四故事女工的 # 定义数据 bob = User(name='bob', birthday='1990-3-21', city='上海', money=370) tom = User(name='tom', birthday='1995-9-12', money=288) lucy = User(name='lucy', birthday='1998-5-14', city='北京') jam = User(name='jam', birthday='1994-3-9', city='深圳', money=86) alex = User(name='alex', birthday='1992-3-17', city='北京') eva = User(name='eva', birthday='1987-7-28', city='深圳', money=631) rob = User(name='rob', birthday='1974-2-5', city='上海', money=735) ella = User(name='ella', birthday='1999-5-26', city='北京') tony = User(name='tony', city='深圳', money=199) # 在数据库中添加数据 session = Session() # 创建会话 session.add_all([bob, tom, lucy, jam, alex, eva, rob, ella, tony]) session.commit()
from main import Session, engine, User
local_session = Session(bind=engine)
user_to_update = local_session.query(User).filter(
User.username == 'jona').first()
user_to_update.username = "******"
user_to_update.email = "*****@*****.**"
local_session.commit()