# create a default user

#u = meeplib.User('test', 'foo')

# create a single message

#meeplib.Message('my title', 'This is my message!', u, "!")

try:

fp = open('users.pickle')

try:

obj = pickle.load(fp)

while True:

#print obj

(a, b) = obj

u = meeplib.User(a, b);

try:

obj = pickle.load(fp)

except EOFError:

break

except EOFError:

pass

except IOError:

fp = open('users.pickle', "w")

fp.close()

try:

fp = open('messages.pickle')

try:

obj = pickle.load(fp)

while True:

(a, b, c, d, e) = obj

#print obj

m = meeplib.Message(a,b,meeplib.get_user(c), d)

m.id = e

#print m.post + " is " + str(m.id)

#obj = pickle.load(fp)

#m.replies = obj

#print obj

try:

obj = pickle.load(fp)

except EOFError:

break

except EOFError:

pass

except IOError:

p = open('messages.pickle', "w")

"""

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', username=username)]

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

#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 = ''

print "USERNAME: " + username + "!"

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

#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):

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)

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(': ')