1 files changed, 115 insertions, 0 deletions
diff --git a/app.py b/app.py
new file mode 100755
index 0000000..b7a62dc
--- /dev/null
+++ b/app.py
@@ -0,0 +1,115 @@
+#! /usr/bin/env python3 -Wall
+import json
+import sqlite3
+import tornado.ioloop
+import tornado.web
+
+conn = sqlite3.connect('example.db')
+c = conn.cursor()
+
+
+def intidb():
+    c.execute("CREATE TABLE albums (id INTEGER PRIMARY KEY AUTOINCREMENT, name text, slug text)")
+    c.execute("CREATE TABLE photos (id INTEGER PRIMARY KEY AUTOINCREMENT, album_id integer, name text, path text)")
+    conn.commit()
+
+
+class Photo:
+    def __init__(self, file):
+        self.path = ''
+
+    def save(self):
+        c.execute("INSERT INTO photos (name")
+
+class Album:
+    def __init__(self, name):
+        self.id = None
+        self.name = name
+
+    @classmethod
+    def get(cls, id):
+        c.execute("SELECT * FROM albums WHERE id = ?", (id,))
+        row = c.fetchone()
+        album = Album(row[1])
+        album.id = row[1]
+        return album
+
+    def save(self):
+        if self.id:
+            c.execute("UPDATE albums SET name = ? WHERE id = ?", (self.name, self.id,))
+        else:
+            c.execute("INSERT INTO albums (name) VALUES (?)", (self.name,))
+            self.id = c.lastrowid
+
+    def get_photos(self):
+        photos = []
+        c.execute("SELECT * FROM photos WHERE album_id = ?", (self.id,))
+        for p in c.fetchall():
+            photos.append(Photo(p))
+
+        return photos
+
+    def asdict(self):
+        return {'id': self.id, 'name': self.name}
+
+    def __str__(self):
+        return self.name
+
+
+class UploadHandler(tornado.web.RequestHandler):
+    def post(self, album):
+        for f in self.request.files['file']:
+            photo = Photo(album_id=album)
+            upload_dir = 'uploads/%d' % album
+            import os
+            try:
+                os.mkdir(upload_dir)
+                fpath = '%s/%s' % (upload_dir, f.filename,)
+                fh = open(fpath, 'wb')
+                fh.write(f.body)
+                photo.save()
+            except Exception, e:
+                print e
+
+
+class MainHandler(tornado.web.RequestHandler):
+    def get(self, id=None):
+        album = Album('New Album')
+        albums = []
+        c.execute('SELECT * FROM albums')
+
+        for r in c.fetchall():
+            album = Album(r[1])
+            album.id = r[0]
+            albums.append(album)
+        
+        self.render("templates/gallery-index.html", albums=albums, album=album)
+
+    def post(self):
+        data = self.request.body.decode('utf-8')
+        data = json.loads(data)
+        album = Album(data['name'])
+        album.save()
+        self.write(album.asdict())
+
+    def put(self, id):
+        data = self.request.body.decode('utf-8')
+        data = json.loads(data)
+        album = Album.get(id)
+        album.name = data['name']
+        album.save()
+
+
+if __name__ == "__main__":
+    #intidb()
+    application = tornado.web.Application([
+        (r"/", MainHandler),
+        (r"/albums/", MainHandler),
+        (r"/albums/(\d+)/", MainHandler),
+        (r"/albums/(\d+)/", MainHandler),
+        (r"/albums/(\d+)/upload/", UploadHandler),
+    ], static_path="static", debug=True)
+
+    application.listen(8888)
+    tornado.ioloop.IOLoop.instance().start()
+    #db.generate_mapping(create_tables=True)