def createUser(login_session):
newUser = User(name=login_session['username'], email=login_session[
'email'], picture=login_session['picture'])
session.add(newUser)
session.commit()
user = session.query(User).filter_by(email=login_session['email']).one()
return user.id
def popUsers():
print "Populating users..."
for user in initData.users:
addUser = User( name=initData.users[user]['name']
, email=initData.users[user]['email'])
session.add(addUser)
session.commit()
print "...done"
async def get_invites():
# TODO: Implement process invites logic
await client.wait_until_ready()
invite_endpoint = DISCORD_API_BASE_URL + '/guilds/%s/invites' % TEST_GUILD_ID
user_endpoint = DISCORD_API_BASE_URL + '/guilds/%s/members' % TEST_GUILD_ID
headers = {'Authorization': 'Bot ' + DISCORD_API_KEY}
while not client.is_closed:
r = requests.get(invite_endpoint, headers=headers)
invites = json.loads(r.text)
for invite in invites:
if invite['channel']['name'] != 'welcome':
continue
user = session.query(User).filter_by(
user_id=invite['inviter']['id']).first()
if not user:
user = User(user_id=invite['inviter']['id'],
count=invite['uses'])
session.add(user)
session.commit()
rank, next = get_role(invite['uses'])
r = requests.get(user_endpoint + '/%s' % invite['inviter']['id'],
headers=headers)
api_user = json.loads(r.text)
if "code" in api_user:
continue
elif "message" in api_user:
if api_user["message"] == "You are being rate limited.":
await asyncio.sleep(api_user["retry_after"] / 1000)
r = requests.get(user_endpoint +
'/%s' % invite['inviter']['id'],
headers=headers)
api_user = json.loads(r.text)
roles = api_user['roles']
user.count = invite['uses']
session.commit()
if rank in roles:
pass
else:
roles.append(rank)
r = requests.patch(user_endpoint +
'/%s' % invite['inviter']['id'],
json={'roles': roles},
headers=headers)
await asyncio.sleep(250) # Check invites every 30 seconds
def add_new_p():
"""Add new patient to data base
Args:
patient_id (key): numerical id of patient
user_age (key) : age of patient
attending_email (key) : patient email
Returns:
result: prints if patient was saved
Returns: if patient added to database """
connect("mongodb://*****:*****@ds157818.mlab.com:57818/hr")
a = request.get_json()
my_id = a["patient_id"]
all_id = []
all_pat = User.objects.raw({})
for user in all_pat:
all_id.append(user.patient_id)
# check if patient exists
check_if_new(all_id, my_id)
# validate correct keys
# try:
# validate_new_patient(a)
# except ValidationError as inst:
# return jsonify({"message": inst.message}),500
patient = User(patient_id=a["patient_id"],
attending_email=a["attending_email"],
user_age=a["user_age"])
patient.save()
logging.info("Added a new patient, %s", a["patient_id"])
result = {"message": "Successfully added new patient"}
return jsonify(result)
def register():
if request.method == "POST":
name = request.form['username']
password = request.form['password']
email = request.form['emailid']
mobile = request.form['Mobilenumber']
new_user = User(name=name,
email=email,
password=password,
mobile=mobile)
session.add(new_user)
session.commit()
return "Successfully Inserted."
else:
return render_template('register.html')
def add_score(user_name, score, items_shot, time_played):
if is_score_eligible(score):
session = Session()
new_user = User(user_name=user_name,
score=score,
items_shot=items_shot,
time_played=time_played)
if get_leaders().count() >= 10:
session.query(User).filter(User.score == _min_score()).delete()
session.add(new_user)
session.commit()
session.close()
else:
raise ScoreNotEligible(
'This score is not good enough to enter the leaders board')
def index():
form = NameForm()
if form.validate_on_submit():
user = User.query.filter_by(username=form.name.data).first()
if user is None:
user = User(username=form.name.data)
db.session.add(user)
db.session.commit()
session['known'] = False
if app.config['FLASKY_ADMIN']:
send_email(app.config['FLASKY_ADMIN'], u'新用户',
'mail/new_user', user=user)
else:
session['known'] = True
session['name'] = form.name.data
return redirect(url_for('index'))
return render_template('index.html', form=form, name=session.get('name'),
known=session.get('known', False))
def register():
if request.form:
username = request.form["username"]
plainpwd = request.form[
"password"] + SALT #Insert salted password hash into db
hashedpwd = hashlib.sha256(plainpwd.encode("utf-8")).hexdigest()
mfa = int(re.sub('[^0-9]', '', request.form["mfa"]))
try: #Record registration time
user = User(username=username, password=hashedpwd, mfa=mfa)
log = Loginlog(username=username, query_type="register")
db.session.add(user)
db.session.add(log)
db.session.commit()
result = "Registration success."
except exc.IntegrityError: #Catch error if username exists in db
result = "Registration failure. Username is already taken."
return render_template('register.html', result=result)
return render_template('register.html')
def add_movie(movie_name, movie_address, user_id):
movie = User(movie_name=movie_name,
movie_address=movie_address,
user_id=user_id)
db.session.add_all([movie])
db.session.commit()
def add_user(user_name, password, userfile):
user = User(user_name=user_name,
user_password=password,
user_file=userfile)
db.session.add_all([user])
db.session.commit()
def write_user(self, user_name, email_address, category):
new_user = User(user_name=user_name,
email=email_address,
category=category)
self.session.add(new_user)
self.session.commit()
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from create_db import Base, Author, Book, User
engine = create_engine('sqlite:///library_catalog.db')
Base.metadata.bind = engine
DBSession = sessionmaker(bind=engine)
session = DBSession()
# Create initial admin user
adminUser = User(
name="admin",
email="*****@*****.**",
picture="anyPicturefound",
)
session.add(adminUser)
session.commit()
# Create authors
auth1 = Author(
user=adminUser,
name="Patrick Rothfuss",
)
session.add(auth1)
session.commit()
auth2 = Author(
name="John Steinbeck",
user=adminUser,
)
