django-restit 4.2.130__tar.gz → 4.2.131__tar.gz

{django_restit-4.2.130 → django_restit-4.2.131}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: django-restit
-Version: 4.2.130
+Version: 4.2.131
 Summary: A Rest Framework for DJANGO
 License: MIT
 Author: Ian Starnes

{django_restit-4.2.130 → django_restit-4.2.131}/medialib/models.py

@@ -330,6 +330,7 @@ class MediaItem(models.Model, RestModel, MetaDataModel):
     Media Item (a video or image)
     """
     class RestMeta:
+        POST_SAVE_FIELDS = ["rendernow"]
         SEARCH_FIELDS = ["name", "description"]
         VIEW_PERMS = ["view_media", "manage_users", "manage_media"]
         SAVE_PERMS = ["manage_media", "manage_users"]
@@ -451,6 +452,12 @@ class MediaItem(models.Model, RestModel, MetaDataModel):
             return self.library.default_store()
         return settings.MEDIALIB_DEFAULT_STORE
 
+    def s3_store(self):
+        return f's3://{settings.AWS_S3_BUCKET}'
+
+    def set_rendernow(self, value):
+        self.new_render()
+
     def uses(self):
         """
         List uses available for this item
@@ -551,6 +558,11 @@ class MediaItem(models.Model, RestModel, MetaDataModel):
         return super(MediaItem, self).save()
 
     def save(self, *args, **kwargs):
+        if self.pk is None:
+            return self.saveNew(*args, **kwargs)
+        return super(MediaItem, self).save(*args, **kwargs)
+
+    def saveNew(self, *args, **kwargs):
         """
         When saving MediaItem set newfile attribute to upload new file and start rendering
         """
@@ -1023,6 +1035,25 @@ class MediaItem(models.Model, RestModel, MetaDataModel):
         media.save()
         return media
 
+    @staticmethod
+    def on_upload_s3(request):
+        filename = request.DATA.get("filename")
+        filesize = request.DATA.get("filesize")
+        filetype = request.DATA.get("filetype")
+        kind = utils.guessMediaKindByName(filename)
+        if kind is None:
+            kind = "*"
+        obj = MediaItem(name=filename, owner=request.member, kind=kind, group=request.group)
+        obj.save()
+        rendition = MediaItemRendition(
+            mediaitem=obj, name="Original", bytes=filesize,
+            use='original', kind=kind, is_original=True)
+        rendition.save()
+        rendition.url = rendition.generateURL(filename, store=obj.s3_store())
+        rendition.save()
+        return dict(url=rendition.generateUploadURL(filetype), id=obj.pk)
+
+
 class MediaItemMetaData(MetaDataBase):
     parent = models.ForeignKey(MediaItem, related_name="properties", on_delete=models.CASCADE)
 
@@ -1073,6 +1104,7 @@ class CuePointMeta(models.Model):
     def __str__(self):
         return self.key
 
+
 class MediaItemRendition(models.Model):
     """
     A rendition of a media item
@@ -1119,20 +1151,18 @@ class MediaItemRendition(models.Model):
         except IndexError:
             return ''
 
-    def upload(self, fp, prefix=""):
-        """
-        Upload rendition to non-volatile storage
-        """
-        doclose = False
-        if type(fp) == str:
-            doclose = True
-            fp = open(fp)
+    def generateUploadURL(self, filetype):
+        from medialib.stores import s3
+        return s3.generate_upload_url(self.url, filetype)
 
+    def generateURL(self, name, prefix="", store=None):
+        if store is None:
+            store = self.mediaitem.default_store()
         paths = []
-        paths.append(self.mediaitem.default_store())
+        paths.append(store)
         paths.append("/{}/".format(int_to_base36(self.mediaitem.pk)))
         path = "".join(paths)
-        if self.rendition_definition != None:
+        if self.rendition_definition is not None:
             paths.append(int_to_base36(self.rendition_definition.pk))
         else:
             paths.append("0")
@@ -1140,13 +1170,24 @@ class MediaItemRendition(models.Model):
         paths.append(utils.toMD5(path, int(time.time())))
         paths.append(".")
         ext = ".dat"
-        if "." in fp.name:
-            ext = fp.name.split('.')[-1]
+        if "." in name:
+            ext = name.split('.')[-1]
         elif hasattr(self.mediaitem, "ext"):
             ext = self.mediaitem.ext
         paths.append(ext)
         # print(paths)
-        self.url = "".join(paths)
+        return "".join(paths)
+
+    def upload(self, fp, prefix=""):
+        """
+        Upload rendition to non-volatile storage
+        """
+        doclose = False
+        if type(fp) == str:
+            doclose = True
+            fp = open(fp)
+
+        self.url = self.generateURL(fp.name, prefix)
         stores.upload(self.url, fp, background=True)
         try:
             self.bytes = fp.size

{django_restit-4.2.130 → django_restit-4.2.131}/medialib/rpc/media.py

@@ -1,4 +1,5 @@
 from rest import decorators as rd
+from rest import views as rv
 from medialib import models as medialib
 
 
@@ -9,6 +10,13 @@ def rest_on_media_item(request, pk=None):
     return medialib.MediaItem.on_rest_request(request, pk)
 
 
+@rd.urlPOST('media/item/s3')
+@rd.login_required
+@rd.requires_params(["filename", "filesize"])
+def rest_on_media_upload_s3(request, pk=None):
+    return rv.restReturn(request, dict(data=medialib.MediaItem.on_upload_s3(request)))
+
+
 @rd.url(r'media/ref/$')
 @rd.url(r'media/ref/(?P<pk>\d+)$')
 @rd.login_required

{django_restit-4.2.130 → django_restit-4.2.131}/medialib/stores/s3.py

@@ -170,6 +170,20 @@ def get_file(url, fp=None):
     return obj.download(fp)
 
 
+def generate_upload_url(url, filetype, expires=3600, acl="public-read"):
+    u = urlparse(url)
+    bucket_name = u.netloc
+    key = u.path.lstrip('/')
+    client = getS3(False)
+    params = dict(Bucket=bucket_name, Key=key, ContentType=filetype)
+    from rest import helpers
+    helpers.log_error("generate_upload_url", params)
+    return client.generate_presigned_url(
+        'put_object',
+        ExpiresIn=expires,
+        Params=params)
+
+
 def delete(url):
     if url[-1] == "/":
         prefix = url.path.lstrip("/")

{django_restit-4.2.130 → django_restit-4.2.131}/medialib/utils.py

@@ -14,7 +14,8 @@ EXT_MAP = {
 KIND_MAP = {
     "video": "V",
     "image": "I",
-    "http": "E"
+    "http": "E",
+    "other": "O"
 }
 
 def getFileExt(file):

{django_restit-4.2.130 → django_restit-4.2.131}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "django-restit"
-version = "4.2.130"
+version = "4.2.131"
 description = "A Rest Framework for DJANGO"
 authors = ["Ian Starnes <ians@311labs.com>"]
 license = "MIT"

{django_restit-4.2.130 → django_restit-4.2.131}/rest/__init__.py

@@ -1,4 +1,4 @@
 from .uberdict import UberDict  # noqa: F401
 from .settings_helper import settings  # noqa: F401
 
-__version__ = "4.2.130"
+__version__ = "4.2.131"

{django_restit-4.2.130 → django_restit-4.2.131}/rest/datem.py

@@ -182,7 +182,9 @@ def getWeek(start, start_day=0):
     return week_start, week_end
 
 
-def getStartOfMonth(d):
+def getStartOfMonth(d, clear_time=False):
+    if clear_time:
+        return d.replace(day=1, hour=0, minute=0, second=0, microsecond=0)
     return d.replace(day=1)
 
 
@@ -190,13 +192,19 @@ def nextMonth(d):
     return getStartOfMonth(getStartOfMonth(d) + timedelta(days=32))
 
 
-def getEndOfMonth(start):
+def getEndOfMonthNoTime(start):
     start = start.replace(day=1, hour=0, minute=0, second=0, microsecond=0)
     end = start + timedelta(days=calendar.monthrange(start.year, start.month)[1])
     end = end.replace(hour=0, minute=0, second=0, microsecond=0)
     return end
 
 
+def getEndOfMonth(d, clear_time=False):
+    if clear_time:
+        return getEndOfMonthNoTime(d)
+    return getStartOfMonth(nextMonth(d)) - timedelta(days=1) 
+
+
 def parseDate(date_str, is_future=False, is_past=False, month_end=True, as_date=False):
     res = parseDateTime(date_str, is_future, is_past, month_end)
     if as_date and res:
@@ -284,10 +292,10 @@ def getDateRange(start, end=None, kind=None, zone=None, hour=0, eod=None, end_eo
             start, end = getWeek(start)
         elif kind == "month":
             start = start.replace(hour=0, day=1)
-            end = getEndOfMonth(start)
+            end = getEndOfMonth(start, True)
         elif kind == "year":
             start = start.replace(hour=0, day=1, month=1)
-            end = getEndOfMonth(start.replace(month=12))
+            end = getEndOfMonth(start.replace(month=12), True)
         elif isinstance(kind, int) or (isinstance(kind, str) and kind.isdigit()):
             end = start + timedelta(days=1)
             start = end - timedelta(days=int(kind))