* server/setup.py: install default spaces.

Author: swannodette

server/setup.py

@@ -49,6 +49,31 @@ def init(dbname="shiftspace/master"):
     sync()
 
 
+def installSpace(space):
+    from models import core
+    import simplejson as json
+    fh = open(os.path.join("spaces", space, "attrs.json"))
+    data = json.loads(fh.read())
+    data["type"] = "space"
+    db = core.connect()
+    if db.get(space):
+        del db[space]
+    db[space] = data
+
+
+DEFAULT_SPACES = (
+    "Notes",
+    "Highlights",
+    "SourceShift",
+    "ImageSwap",
+    )
+
+
+def installDefaultSpaces():
+    for space in DEFAULT_SPACES:
+        installSpace(space)
+
+
 def sync(createAdmin=True):
     import models.core as core
     from models.ssuser import SSUser
@@ -68,6 +93,8 @@ def sync(createAdmin=True):
         master["admins"] = adminDoc
         master["shiftspace"] = adminUser
 
+    installDefaultSpaces()
+
     Spaces = ViewDefinition('spaces', 'by_name', '''
        function(doc) {             
          if(doc.type == "space") { 

shifty.py

@@ -379,12 +379,7 @@ def installSpace(space):
     """
     Installs a space's attributes into the database.
     """
-    from server.models import core
-    fh = open(os.path.join("spaces", space, "attrs.json"))
-    data = json.loads(fh.read())
-    data["type"] = "space"
-    db = core.connect()
-    db[space] = data
+    setup.installSpace(space)
 
 
 def shell():