Completed
Push — master ( d1e883...1f56b4 )
by
unknown
01:28
created

pre_sparkle_save()   A

Complexity

Conditions 3

Size

Total Lines 9

Duplication

Lines 0
Ratio 0 %

Importance

Changes 1
Bugs 0 Features 0
Metric Value
cc 3
c 1
b 0
f 0
dl 0
loc 9
rs 9.6666
1
# coding: utf8
2
3
"""
4
This software is licensed under the Apache 2 license, quoted below.
5
6
Copyright 2014 Crystalnix Limited
7
8
Licensed under the Apache License, Version 2.0 (the "License"); you may not
9
use this file except in compliance with the License. You may obtain a copy of
10
the License at
11
12
    http://www.apache.org/licenses/LICENSE-2.0
13
14
Unless required by applicable law or agreed to in writing, software
15
distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
16
WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
17
License for the specific language governing permissions and limitations under
18
the License.
19
"""
20
21
from __future__ import unicode_literals
22
from django.utils.encoding import python_2_unicode_compatible
23
24
import os
25
26
from django.db import models
27
from django.dispatch import receiver
28
from django.db.models.signals import pre_delete,pre_save
29
30
from omaha.models import BaseModel, Application, Channel
31
from sparkle.managers import VersionManager
32
33
34
def version_upload_to(obj, filename):
35
    return os.path.join('sparkle', obj.app.name, obj.channel.name, str(obj.version), filename)
36
37
38
@python_2_unicode_compatible
39
class SparkleVersion(BaseModel):
40
    is_enabled = models.BooleanField(default=True)
41
    app = models.ForeignKey(Application)
42
    channel = models.ForeignKey(Channel, db_index=True)
43
    version = models.CharField(max_length=32)
44
    short_version = models.CharField(max_length=32, blank=True, null=True)
45
    release_notes = models.TextField(blank=True, null=True)
46
    file = models.FileField(upload_to=version_upload_to, null=True)
47
    file_size = models.PositiveIntegerField(null=True, blank=True)
48
    dsa_signature = models.CharField(verbose_name='DSA signature', max_length=140, null=True, blank=True)
49
50
    objects = VersionManager()
51
52
    class Meta:
53
        index_together = (
54
            ('app', 'channel'),
55
        )
56
        unique_together = (
57
            ('app', 'channel', 'version'),
58
        )
59
60
    def __str__(self):
61
        return "{app} {version}".format(app=self.app, version=self.version)
62
63
    @property
64
    def file_absolute_url(self):
65
        return self.file.url
66
67
    @property
68
    def file_package_name(self):
69
        return os.path.basename(self.file_absolute_url)
70
71
    @property
72
    def file_url(self):
73
        return '%s/' % os.path.dirname(self.file_absolute_url)
74
75
    @property
76
    def size(self):
77
        return self.file_size
78
79
80
@receiver(pre_save, sender=SparkleVersion)
81
def pre_sparkle_save(sender, instance, *args, **kwargs):
82
    if instance.pk:
83
        old = sender.objects.get(pk=instance.pk)
84
        if old.file == instance.file:
85
            return
86
        else:
87
            old.file.delete(save=False)
88
            old.file_size = 0
89
90
91
@receiver(pre_delete, sender=SparkleVersion)
92
def pre_version_delete(sender, instance, **kwargs):
93
    storage, name = instance.file.storage, instance.file.name
94
    if name:
95
        storage.delete(name)
96