1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
# coding: utf8
"""
This software is licensed under the Apache 2 license, quoted below.
Copyright 2015 Crystalnix Limited
Licensed under the Apache License, Version 2.0 (the "License"); you may not
use this file except in compliance with the License. You may obtain a copy of
the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
License for the specific language governing permissions and limitations under
the License.
"""
import os
import uuid
from django.db import models
from django.db.models.signals import pre_delete
from django.dispatch import receiver
from django.utils import timezone
from jsonfield import JSONField
from omaha.models import BaseModel
from feedback.managers import FeedbackManager
def upload_to(directory, obj, filename):
now = timezone.now()
max_length = 100
path = os.path.join(*map(str, [directory, now.year, now.month,
now.day, uuid.uuid4(), filename]))
if len(path) > max_length:
name, ext = os.path.splitext(path)
ext_length = len(ext)
path = name[:max_length-ext_length] + ext
return path
def screenshot_upload_to(obj, filename):
return upload_to('screenshot', obj, filename)
def blackbox_upload_to(obj, filename):
return upload_to('blackbox', obj, filename)
def logs_upload_to(obj, filename):
return upload_to('system_logs', obj, filename)
def attach_upload_to(obj, filename):
return upload_to('feedback_attach', obj, filename)
class Feedback(BaseModel):
description = models.TextField()
email = models.CharField(max_length=500, null=True, blank=True)
page_url = models.CharField(max_length=2048, null=True, blank=True)
screenshot = models.ImageField(upload_to=screenshot_upload_to, blank=True, null=True)
screenshot_size = models.PositiveIntegerField(null=True, blank=True)
blackbox = models.FileField(upload_to=blackbox_upload_to, blank=True, null=True)
blackbox_size = models.PositiveIntegerField(null=True, blank=True)
system_logs = models.FileField(upload_to=logs_upload_to, blank=True, null=True)
system_logs_size = models.PositiveIntegerField(null=True, blank=True)
attached_file = models.FileField(upload_to=attach_upload_to, blank=True, null=True)
attached_file_size = models.PositiveIntegerField(null=True, blank=True)
feedback_data = JSONField(verbose_name='Feedback data', help_text='JSON format', null=True, blank=True)
ip = models.GenericIPAddressField(blank=True, null=True)
created_at = models.DateTimeField(auto_now_add=True, db_index=True)
updated_at = models.DateTimeField(auto_now=True)
objects = FeedbackManager()
@property
def size(self):
return self.screenshot_size + self.blackbox_size + self.system_logs_size + self.attached_file_size
@receiver(pre_delete, sender=Feedback)
def pre_feedback_delete(sender, instance, **kwargs):
file_fields = [instance.screenshot, instance.blackbox, instance.system_logs, instance.attached_file]
for field in file_fields:
storage, name = field.storage, field.name
if name:
storage.delete(name)
class FeedbackDescription(Feedback):
class Meta:
proxy = True
|
