From 0528359ce878f9b383ce3b461b72bb4196e257aa Mon Sep 17 00:00:00 2001
From: TheOrangeOne <git@theorangeone.net>
Date: Mon, 5 Sep 2016 18:53:18 +0100
Subject: [PATCH] create basic model

---
 project/files/models.py | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 project/files/models.py

diff --git a/project/files/models.py b/project/files/models.py
new file mode 100644
index 0000000..805f32a
--- /dev/null
+++ b/project/files/models.py
@@ -0,0 +1,30 @@
+from django.db import models
+import os
+import uuid
+from admin_resumable.fields import ModelAdminResumableFileField
+from shortuuidfield import ShortUUIDField
+
+
+def build_save_path(obj, filename):
+    return obj.get_save_path(filename)
+
+
+class SharedFile(models.Model):
+    id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
+    short_id = ShortUUIDField()
+    created = models.DateTimeField(auto_now_add=True)
+    file = ModelAdminResumableFileField(upload_to=build_save_path)
+    hotlink = models.BooleanField(default=False)
+    published = models.BooleanField(default=True)
+
+    filename = ""
+
+    def get_save_path(self, filename):
+        self.filename = filename
+        return os.path.join(str(self.id), str(self.short_id), filename)
+
+    def get_private_path(self):
+        return self.get_save_path(self.filename)
+
+    def get_original_filename(self):
+        return "_".join(self.file.name.split('_')[1:])