Make User.name unique.

[zzz-floof.git] / floof / websetup.py

diff --git a/floof/websetup.py b/floof/websetup.py
index b1611c9..a4e935f 100644 (file)
--- a/floof/websetup.py
+++ b/floof/websetup.py
@@ -10,23 +10,40 @@ from pylons import config
 from elixir import *
 from floof import model as model
 
 from elixir import *
 from floof import model as model
 

+import os
+

 def setup_app(command, conf, vars):
     """Place any commands to setup floof here"""
     load_environment(conf.global_conf, conf.local_conf)
 def setup_app(command, conf, vars):
     """Place any commands to setup floof here"""
     load_environment(conf.global_conf, conf.local_conf)

-    model.metadata.create_all()

 
 

-    # Initialisation here ... this sort of stuff:
+    ### Database schema
+    model.metadata.create_all()

 
 

+    ### Sample data

     # Users
     from floof.model.users import IdentityURL, User
     identity_url = IdentityURL(url=u'http://eevee.livejournal.com/')
     # Users
     from floof.model.users import IdentityURL, User
     identity_url = IdentityURL(url=u'http://eevee.livejournal.com/')

-    user = User(name=u'Eevee')
+    user = User(name=u'eevee')

     user.identity_urls.append(identity_url)
 
     user.identity_urls.append(identity_url)
 

-
-    # some_entity = model.Session.query(model.<modelfile>.<Some_Entity>).get(1)
-    # e.g. foo = model.Session.query(model.identity.User).get(1)
-    # from datetime import datetime
-    # some_entity.poked_on = datetime.now()
-    # model.Session.add(some_entity)

     model.Session.commit()
     model.Session.commit()

+
+
+    ### Filesystem stuff
+    # Only necessary if we're using the filesystem for storage.
+    # And we are!
+    art_path = os.path.join(config['app_conf']['static_root'], 'art')
+    # Test that we can write to the art directory
+    if os.access(art_path, os.W_OK | os.X_OK):
+        # Create directories for various image sizes
+        for subdir in ['original', 'medium', 'thumbnail']:
+            path = os.path.join(art_path, subdir)
+            # If it exists, it should be writeable.  If not, create it.
+            if os.path.exists(path):
+                if not os.access(path, os.W_OK | os.X_OK):
+                    print "*** WARNING: Can't write to the art/{0} " \
+                          "directory!".format(subdir)
+                continue
+            os.makedirs(path)
+    else:
+        print "*** WARNING: Can't write to the art directory!"