def run(): form = CreateUserForm() if form.validate_on_submit(): print("name=" + form.name.data + "\n" + "user_id=" + form.user_id.data + "\n" + "email=" + form.email.data + "\n") users.create_user(name=form.name.data, user_id=form.user_id.data, email=form.email.data) # TODO: Check and return errors. return redirect("/index") display = {} display["form"] = form return render_template("create_user.html", display=display)
def post(self): connection = MySQLdb.connect(host=self.get_argument('mysql_host'), user=self.get_argument('mysql_user'), passwd=self.get_argument('mysql_password'), db=self.get_argument('mysql_database'), charset="utf8", use_unicode=True) cursor = connection.cursor() initial_sql = open(os.path.normpath(os.path.realpath(__file__) + '/../../models/helloworld.sql')) cursor.execute(initial_sql.read()) cursor.close() self.constants['mysql_host'] = self.get_argument('mysql_host') self.constants['mysql_user'] = self.get_argument('mysql_user') self.constants['mysql_password'] = self.get_argument('mysql_password') self.constants['mysql_database'] = self.get_argument('mysql_database') self.get_db_connection() user = self.models.users.get(1) if not user: user = users.create_user(self, self.get_argument('username'), self.get_argument('email')) user.author = True user.superuser = True user.save() users.create_empty_content(self, user.username, 'main', 'main', "Hello, world.", view="This is a brand-spankin' new Hello, world site.") # write site.cfg config = ConfigParser.RawConfigParser() config.add_section('general') for k, v in constants.defaults.items(): config.set('general', k, v) if self.request.protocol == 'http': config.set('general', 'http_prefix', self.get_argument('prefix')) config.set('general', 'https_prefix', '') else: config.set('general', 'https_prefix', self.get_argument('prefix')) config.set('general', 'http_prefix', '') config.set('general', 'mysql_database', self.get_argument('mysql_database')) config.set('general', 'mysql_host', self.get_argument('mysql_host')) config.set('general', 'mysql_user', self.get_argument('mysql_user')) config.set('general', 'mysql_password', self.get_argument('mysql_password')) xsrf_secret = ''.join([choice(string.letters) for i in range(50)]) xsrf_secret = hashlib.sha256(xsrf_secret).hexdigest() config.set('general', 'xsrf_secret', xsrf_secret) with open(self.application.settings["config_path"], 'w') as configfile: config.write(configfile)
def test_create_user(self): user_key = users.create_user("foo", "1455", "*****@*****.**") self.assertTrue(user_key != None) user = users.get_user("1455") self.assertTrue(user.name == "foo") self.assertTrue(user.user_id == "1455") self.assertTrue(user.email == "*****@*****.**")
def post(self): if not self.authenticate(superuser=True): return content_url = url_factory.load_basic_parameters(self, prefix="/users") user = users.create_user(self, content_url["profile"], self.get_argument('email')) self.save_user(user) self.set_status(204)
def run(): google_user = api.users.get_current_user() if google_user is None: flask.flash(u'You denied the request to sign in.') return flask.redirect("/index") google_id = str(google_user.user_id()) name = google_user.nickname() email = google_user.email() luser = users.get_user_by_google_id(google_id) if not luser: luser = users.create_user(name, email, google_id=google_id) if login.login_user(luser): flask.get_flashed_messages() return flask.redirect(request.args.get("next") or flask.url_for("index")) else: flask.flash("Login Fail") return flask.redirect(flask.url_for("index"))
def test_all(self): user_key = users.create_user("foo", "1455", "*****@*****.**") self.assertTrue(user_key != None) timeline_key = eventlines.create_timeline("1455", "timeline_foo", "description for timeline") timeline_id = timeline_key.id() timeline = eventlines.get_timeline(timeline_id) self.assertTrue(timeline.name == "timeline_foo") self.assertTrue(timeline.user_id == "1455") self.assertTrue(timeline.description == "description for timeline") user = users.get_user("1455") self.assertTrue(user.timeline_ids == [timeline_id]) key = eventlines.add_event( "1455", timeline_id, "event1", "event desc", datetime.datetime.strptime("2014/03/01", "%Y/%m/%d"), datetime.datetime.strptime("2014/03/01", "%Y/%m/%d")) self.assertTrue(key.id() == timeline_id) timeline = eventlines.get_timeline(timeline_id) self.assertTrue(timeline.events[0].name == "event1")