aboutsummaryrefslogtreecommitdiffstats
path: root/piccast/feeds/models.py
blob: 2b04b7a94a8423a2a106dfdd8b68a7cb6397e4e1 (plain)
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
from django.db import models

class Category(models.Model):
    #id
    name = models.CharField(max_length=50);

    def __unicode__(self):
        return self.name
    
    def serializable(self):
        return {'id': self.id, 
                'name': self.name }

class Pic(models.Model):
    #id
    set = models.ForeignKey('feeds.PicSet', null=True,blank=True)
    title = models.CharField(max_length=128, blank=False)
    thumbnail_url  = models.URLField(blank=True)
    thumbnail_height = models.PositiveIntegerField(null=True,blank=True)
    thumbnail_width = models.PositiveIntegerField(null=True,blank=True)
    original_url = models.URLField(blank=False)
    original_height = models.PositiveIntegerField(null=True,blank=True)
    original_width = models.PositiveIntegerField(null=True,blank=True)
    source_url = models.URLField(blank=True)
    caption = models.TextField(blank=True)
    credits = models.TextField(blank=True)
    is_nsfw  = models.NullBooleanField(null=True, blank=True)
    
    def __unicode__(self):
        return self.title
    
    def serializable(self):
        return {'id': self.id,
                'set': {'id': self.set.id, 'title': self.set.title },
                'thumbnail_url': self.thumbnail_url,
                'thumbnail_height': self.thumbnail_height,
                'thumbnail_width': self.thumbnail_width,
                'original_url': self.original_url,
                'original_height': self.original_height,
                'original_width': self.original_width,
                'source_url': self.source_url,
                'caption': self.caption,
                'credits': self.credits, 
                'is_nsfw': self.is_nsfw, 
                }
    
    def serializable_short(self):
        # TODO: more here?
        return {'id': self.id, 
                'original_url': self.original_url,
                'original_height': self.original_height,
                'original_width': self.original_width,
                'credits': self.credits,
                } 

class PicFeed(models.Model):
    #id
    created = models.DateTimeField()
    shortname = models.SlugField(blank=False)
    source_url = models.URLField(blank=True)
    rssfeed_url = models.URLField(blank=True)
    favicon_url = models.URLField(blank=True)
    is_active = models.BooleanField(default=False)
    title  = models.CharField(max_length=80, blank=True)
    description = models.TextField(blank=True)
    image  = models.ForeignKey('feeds.Pic', blank=True, null=True)
    category = models.ForeignKey('feeds.Category', null=True,blank=True)

    def __unicode__(self):
        return self.shortname
    
    def serializable(self):
        return {'id': self.id,
                 'created_unixtime': 
                  (int(self.created.strftime("%s")) if self.created else None),
                'shortname': self.shortname,
                'source_url': self.source_url,
                'rssfeed_url': self.rssfeed_url,
                'favicon_url': self.favicon_url,
                'title': self.title,
                'description': self.description, 
                'image': 
                  (self.image.serializable_short() if self.image else None),
                'category': 
                  (self.category.name if self.category else None),
                }

class PicSet(models.Model):
    #id
    created = models.DateTimeField(null=True,blank=True)
    source_url = models.URLField(blank=True)
    title = models.CharField(max_length=128, blank=False)
    description = models.TextField(blank=True)
    credits = models.TextField(blank=True)
    keywords = models.CharField(max_length=256, blank=True)
    image = models.ForeignKey('feeds.Pic', null=True,blank=True)
    feed = models.ForeignKey('feeds.PicFeed', null=True,blank=False)
    category = models.ForeignKey('feeds.Category', null=True,blank=True)
    is_nsfw = models.BooleanField(default=False, blank=False);
    is_valid = models.BooleanField(default=True);

    def __unicode__(self):
        return self.title
    
    def serializable(self):
        return {'id': self.id,
                'created_unixtime': 
                  (int(self.created.strftime("%s")) if self.created else None),
                'source_url': self.source_url,
                'title': self.title,
                'description': self.description, 
                'credits': self.credits, 
                'keywords': self.keywords, 
                'image': 
                  (self.image.serializable_short() if self.image else None),
                'pic_count':
                  Pic.objects.filter(set=self.id).count(),
                'category': 
                  (self.category.name if self.category else None),
                }