if(len(get_all_users()) == 0):

u = User('test', 'foo')

#Create Tables!

conn = sqlite3.connect('meep.db')

c = conn.cursor()

try:

c.execute('''CREATE TABLE Users(username text, password text)''')

except:

pass

try:

c.execute('''CREATE TABLE Message(title text, post text, author text, pid text)''')

except:

#c.execute("TRUNCATE TABLE Message")

pass

conn.commit()

#"""

#WSGI app object.

#"""

def index(self, environ, start_response):

start_response("200 OK", [('Content-type', 'text/html')])

username = 'test'

s = environ.get('HTTP_COOKIE', '')

#print "First Cookie Info: "

#print s

return [render_page('login.html')]

def main_page(self, environ, start_response):

#try:

# meeplib.get_curr_user()

#except NameError:

# meeplib.delete_curr_user()

headers = [('Content-type', 'text/html')]

start_response("200 OK", headers)

username = meeplib.get_curr_user()

return [render_page('index.html')]

def create_user(self, environ, start_response):

headers = [('Content-type', 'text/html')]

start_response("302 Found", headers)

return render_page('create_user.html')

def add_new_user(self, environ, start_response):

#print environ['wsgi.input']

form = cgi.FieldStorage(fp=environ['wsgi.input'], environ=environ)

returnStatement = "user added"

try:

username = form['username'].value

except KeyError:

username = None

try:

password = form['password'].value

except KeyError:

password = None

conn = sqlite3.connect('meep.db')

c = conn.cursor()

if username and password:

c.execute("INSERT INTO Users VALUES('" + username + "', '" + password + "')")

conn.commit()

#print username

#print password

# Test whether variable is defined to be None

if username is None:

returnStatement = "username was not set. User could not be created"

if password is None:

returnStatement = "password was not set. User could not be created"

else:

new_user = meeplib.User(username, password)

SaveUsers()

headers = [('Content-type', 'text/html')]

headers.append(('Location', '/'))

start_response("302 Found", headers)

return [returnStatement]

def login(self, environ, start_response):

print environ['wsgi.input']

form = cgi.FieldStorage(fp=environ['wsgi.input'], environ=environ)

returnStatement = "logged in"

try:

username = form['username'].value

except KeyError:

username = None

try:

password = form['password'].value

except KeyError:

password = None

k = ''

v = ''

conn = sqlite3.connect('meep.db')

c = conn.cursor()

if username and password:

c.execute("SELECT * FROM Users WHERE username = '" + username + "' AND password = '" + password + "'")

if c.fetchone():

k = 'Location'

v = '/main_page'

returnStatement = "logged in"

meeplib.set_curr_user(username)

else:

k = 'Location'

v = '/'

returnStatement = """<p>Invalid user. Please try again.</p>"""

# Test whether variable is defined to be None

#if username is not None:

# if password is not None:

# if meeplib.check_user(username, password) is False:

# k = 'Location'

# v = '/'

# returnStatement = """<p>Invalid user. Please try again.</p>"""

#

# else:

# new_user = meeplib.User(username, password)

# meeplib.set_curr_user(username)

# k = 'Location'

# v = '/main_page'

# else:

# returnStatement = """<p>password was not set. User could not be created</p>"""

#else:

# returnStatement = """<p>username was not set. User could not be created</p>"""

#print """isValidafter: %s """ %(meeplib.check_user(username, password),)

# set content-type

headers = [('Content-type', 'text/html')]

if returnStatement is "logged in":

c = SimpleCookie()

cookie_name, cookie_val = make_set_cookie_header('username', username)

headers.append((cookie_name, cookie_val))

print cookie_name + cookie_val

headers.append((k, v))

start_response('302 Found', headers)

print "everything seems to be working up till now..."

#return self.main_page(environ, start_response)

return returnStatement

def logout(self, environ, start_response):

# does nothing

headers = [('Content-type', 'text/html')]

# send back a redirect to '/'

k = 'Location'

v = '/'

headers.append((k, v))

start_response('302 Found', headers)

return "no such content"

def list_messages(self, environ, start_response):

user = meeplib.get_curr_user();

meeplib._reset()

meeplib.set_curr_user(user)

conn = sqlite3.connect('meep.db')

c = conn.cursor()

for row in c.execute("SELECT * FROM Message"):

print row

meeplib.Message(row[0], row[1], row[2], row[3])

messages = meeplib.get_all_messages()

headers = [('Content-type', 'text/html')]

start_response("200 OK", headers)

return [render_page('list_messages.html', messages=messages)]

def add_message(self, environ, start_response):

form = cgi.FieldStorage(fp=environ['wsgi.input'], environ=environ)

pID = form['pID'].value

headers = [('Content-type', 'text/html')]

parentTitle = ""

start_response("200 OK", headers)

if(pID != '!'):

parentTitle = "RE: " + meeplib.get_message(int(pID)).title

return render_page('add_message.html', pID = pID, parentTitle = parentTitle);

def add_message_action(self, environ, start_response):

print environ['wsgi.input']

form = cgi.FieldStorage(fp=environ['wsgi.input'], environ=environ)

title = form['title'].value

message = form['message'].value

pID = form['pID'].value

username = meeplib.get_curr_user()

user = meeplib.get_user(username)

conn = sqlite3.connect('meep.db')

c = conn.cursor()

c.execute("INSERT INTO Message VALUES('" + title + "', '" + message + "', '" + username + "', '" + pID + "')")

conn.commit()

#new_message = meeplib.Message(title, message, user, pID)

headers = [('Content-type', 'text/html')]

headers.append(('Location', '/m/list'))

start_response("302 Found", headers)

#SaveMessages()

if(pID != '!'):

return ["Message Added"]

else:

return ["Reply Added"]

def del_message(self, environ, start_response):

#print environ['wsgi.input']

form = cgi.FieldStorage(fp=environ['wsgi.input'], environ=environ)

#print form

messages = meeplib.get_all_messages()

s = "Yup"

found = False

for m in messages:

if m.id == int(form['id'].value):

if meeplib.get_curr_user() != m.author.username:

s= "Nope"

break

if m.pID != "!":

meeplib.delete_reply(m)

meeplib.delete_message(m)

found = True

if found == True:

break

start_response("200 OK", [('Content-type', 'text/html')])

#s.append("<p><p><a href = '/m/list'>Return to Messages</a>")

SaveMessages()

return render_page('del_message.html', s = s)

def __call__(self, environ, start_response):

# store url/function matches in call_dict

call_dict = { '/': self.index,

'/main_page': self.main_page,

'/create_user': self.create_user,

'/add_new_user':self.add_new_user,

'/login': self.login,

'/logout': self.logout,

'/m/list': self.list_messages,

'/m/add': self.add_message,

'/m/add_action': self.add_message_action,

'/m/delete': self.del_message

}

# see if the URL is in 'call_dict'; if it is, call that function.

url = environ['PATH_INFO']

fn = call_dict.get(url)

was_picture = False

if(url.startswith('/files/')):

fn = file_server.file_server(url[len('/files/'):])

was_picture = True

if fn is None and not was_picture:

start_response("404 Not Found", [('Content-type', 'text/html')])

return ["Page not found."]

try:

return fn(environ, start_response)

except:

tb = traceback.format_exc()

x = "<h1>Error!</h1><pre>%s</pre>" % (tb,)

status = '500 Internal Server Error'

start_response(status, [('Content-type', 'text/html')])

if(level != 0):

s.append('<blockquote>')

s.append('id: %d<br>' % (m.id,))

s.append('title: %s<br>' % (m.title))

s.append('message: %s<br>' % (m.post))

s.append('author: %s' % (m.author.username))

s.append("""

<form action = 'add' method = 'POST' style="margin:0;">

<input type = 'submit' value = 'Reply' />

<input type = 'hidden' name = 'pID' value = '%d' />

</form>""" % (m.id))

s.append("""

<form action = 'delete' method = 'POST' style="margin:0;">

<input type = 'submit' value = 'Delete Post' />

<input type = 'hidden' name = 'id' value = '%d' />

</form>

<hr>""" % (m.id))

if(m.replies != []):

print m.replies

for r in m.replies:

print_messages(meeplib.get_message(r), s, level + 1)

s.append('</blockquote>')

#Save Messages

#

filename = 'messages.pickle'

fp = open(filename, 'w')

m = meeplib.get_all_messages()

for d in m:

f = [d.title, d.post, d.author.username, d.pID, d.id]

pickle.dump(f, fp)

#f = d.replies

#pickle.dump(f, fp)

fp.close()

#

#Save Users

#

filename = 'users.pickle'

fp = open(filename, 'w')

u = meeplib.get_all_users()

for d in u:

f = [d.username, d.password]

#print f

pickle.dump(f, fp)

fp.close()

"""

Makes a 'Set-Cookie' header.

"""

c = SimpleCookie()

c[name] = value

c[name]['path'] = path

# can also set expires and other stuff. See

# Examples under http://docs.python.org/library/cookie.html.

s = c.output()

(key, value) = s.split(': ')