goodylabs internship exercise 1

blog.py

@@ -0,0 +1,61 @@
+#!/usr/bin/python
+#coding:utf8
+from flask import Flask, render_template, url_for, request,redirect,make_response,session
+import os, MySQLdb
+app = Flask(__name__)
+app.secret_key= 'afjlsjfowflajflkajfkjfkaljf'
+user_list = ['jim', 'max', 'py']
+imagepath = os.path.join(os.getcwd(), "static/images")
+@app.route('/')
+def index():
+    username = request.cookies.get('username')
+    if not username:
+        username = u'Please log in'
+    islogin = session.get('islogin')
+    nav_list = [u'home',u'circles',u'settings']
+    blog = {'title':'welcome to my homepage','content':'hello, nice to see you.'}
+    blogtag = {'javascript':10,"python":20,"shelling":5}
+    img = url_for('static', filename="images/cat.jpg")
+    return render_template('index.html', nav_list=nav_list, username=username, blog = blog, blogtag = blogtag, img=img, islogin=islogin)
+@app.route('/reg', methods=['GET','POST'])
+def regist():
+    if request.method == 'POST':
+        username = request.form['username']
+        conn = MySQLdb.connect(user='root',passwd='admin',host='127.0.0.1')
+        conn.select_db('blog')
+        curr = conn.cursor()
+        sql = 'insert into `user` (`id`,`username`) values (%d,"%s")' % (1,username)
+        curr.execute(sql)
+        conn.commit()
+        curr.close()
+        conn.close()
+        return "user %s regist ok!" % request.form['username']
+    else:
+        #request.args['username']
+        return render_template('regist.html')
+@app.route('/upload', methods=['GET','POST'])
+def upload():
+    if request.method == 'POST':
+        username = request.form['username']
+        file = request.files['img']
+        filename = file.filename
+        file.save(os.path.join(imagepath,filename))
+        return "<img src='static/images/%s' alt=''/>" % filename
+    else:
+        return render_template('upload.html')
+@app.route('/login/', methods=['GET','POST'])
+def login():
+    if request.method == 'POST':
+        username = request.form.get('username')
+        if username in user_list:
+            response = make_response(redirect('/'))
+            response.set_cookie('username', value=username, max_age=300)
+            session['islogin'] = '1'
+            return response
+        else:
+            session['islogin'] = '0'
+            return redirect('/login/')
+    else:
+        return render_template('login.html')
+if __name__ == '__main__':
+    app.run(debug=True,host='0.0.0.0',port=5000)