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, )