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