Flesh out the create API and wire everything up together. Next up, testing.

data/database.py

@@ -151,10 +151,11 @@ class RepositoryTag(BaseModel):
 
 
 class RepositoryBuild(BaseModel):
+  repository = ForeignKeyField(Repository)
+  resource_key = CharField()
   digitalocean_build_node_id = IntegerField(null=True)
   phase = CharField(default='waiting')
   status_url = CharField(null=True)
-  repository = ForeignKeyField(Repository)
 
 
 class QueueItem(BaseModel):

data/model.py

@@ -287,8 +287,8 @@ def set_repository_visibility(repo, visibility):
   repo.save()
 
 
-def create_repository(namespace, name, owner):
-  private = Visibility.get(name='private')
+def create_repository(namespace, name, owner, visibility='private'):
+  private = Visibility.get(name=visibility)
   repo = Repository.create(namespace=namespace, name=name,
                            visibility=private)
   admin = Role.get(name='admin')
@@ -560,3 +560,7 @@ def get_repository_build(request_dbid):
   except RepositoryBuild.DoesNotExist:
     msg = 'Unable to locate a build by id: %s' % request_dbid
     raise InvalidRepositoryBuildException(msg)
+
+
+def create_repository_build(repo, resource_key):
+  return RepositoryBuild.create(repository=repo, resource_key=resource_key)

data/userfiles.py

@@ -0,0 +1,34 @@
+import boto
+import os
+
+from boto.s3.key import Key
+from uuid import uuid4
+
+
+class S3FileWriteException(Exception):
+  pass
+
+
+class UserRequestFiles(object):
+  def __init__(self, s3_access_key, s3_secret_key, bucket_name):
+    self._s3_conn = boto.s3.connection.S3Connection(s3_access_key,
+                                                    s3_secret_key,
+                                                    is_secure=False)
+    self._bucket = self._s3_conn.get_bucket(bucket_name)
+    self._prefix = 'userfiles'
+
+  def store_file(self, flask_file):
+    file_id = str(uuid4())
+    full_key = os.path.join(self._prefix, file_id)
+    k = Key(full_key)
+    bytes_written = k.set_contents_from_file(flask_file)
+
+    if bytes_written == 0:
+      raise S3FileWriteException('Unable to write file to S3')
+
+    return file_id
+
+  def get_file_url(self, file_id, expires_in=300):
+    full_key = os.path.join(self._prefix, file_id)
+    k = Key(full_key)
+    return k.generate_url(expires_in)