init

files_widget/models.py

@@ -0,0 +1,55 @@
+from django.db import models
+from django.contrib.contenttypes.models import ContentType
+from django.contrib.auth.models import Group, Permission
+
+from .fields import ImageField
+
+
+class GlobalPermissionManager(models.Manager):
+    def get_query_set(self):
+        return super(GlobalPermissionManager, self).\
+            get_query_set().filter(content_type__name='global_permission')
+
+
+class GlobalPermission(Permission):
+    """A global permission, not attached to a model"""
+
+    objects = GlobalPermissionManager()
+
+    class Meta:
+        proxy = True
+
+    def save(self, *args, **kwargs):
+        ct, created = ContentType.objects.get_or_create(
+            name="global_permission", app_label=self._meta.app_label
+        )
+        self.content_type = ct
+        super(GlobalPermission, self).save(*args, **kwargs)
+
+
+try:
+    permission = GlobalPermission.objects.get_or_create(
+        codename='can_upload_files',
+        name='Can Upload Files',
+    )
+except:
+    # "Table 'fileswidgettest16.auth_permission' doesn't exist"
+    # it should exist the next time that this file is loaded
+    pass
+
+
+class IconSet(models.Model):
+    name = models.CharField(max_length=50, unique=True)
+    css_path = models.CharField(max_length=200, blank=True, null=True, help_text='Optional css file for icon styling')
+    active = models.BooleanField(default=True)
+    priority = models.IntegerField(default=1)
+    default_icon = models.ForeignKey('files_widget.FileIcon', null=True, blank=True, on_delete=models.SET_NULL)
+
+
+class FileIcon(models.Model):
+    icon_set = models.ForeignKey('files_widget.IconSet', on_delete=models.CASCADE)
+    extension = models.CharField(max_length=100, blank=True, null=True)
+    image = ImageField()
+    display_text_overlay = models.BooleanField(default=True)
+    overlay_text = models.CharField(max_length=7, blank=True, null=True, help_text='Leave blank to display file extension')
+    base_color = models.CharField(max_length=12, blank=True, null=True)