Thomas Peetz
122 lines
3.0 KiB
Python
122 lines
3.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
from mongoengine import connect, Document
|
|
from mongoengine import StringField, BooleanField
|
|
from mongoengine import ListField, ReferenceField, GenericReferenceField
|
|
import pprint
|
|
|
|
|
|
class Publisher(Document):
|
|
name = StringField(required=True)
|
|
|
|
def __str__(self):
|
|
s = "Publisher(%s)" % self.name
|
|
return s
|
|
|
|
|
|
class Artist(Document):
|
|
name = StringField(required=True)
|
|
className = StringField()
|
|
|
|
def __str__(self):
|
|
s = "Artist(%s)" % self.name
|
|
return s
|
|
|
|
|
|
class Issue(Document):
|
|
number = StringField()
|
|
comic = GenericReferenceField()
|
|
is_read = BooleanField(default=False)
|
|
is_stock = BooleanField(default=False)
|
|
|
|
def __str__(self):
|
|
s = "Issue(%s # %s, %s)" % (self.comic.title, self.number, self.is_read)
|
|
return s
|
|
|
|
|
|
class StoryArc(Document):
|
|
name = StringField(required=True)
|
|
comic = GenericReferenceField()
|
|
issues = ListField(ReferenceField(Issue))
|
|
|
|
def __str__(self):
|
|
s = "StoryArc(%s, %s)" % (self.name, self.comic.title)
|
|
return s
|
|
|
|
|
|
class Volume(Document):
|
|
name = StringField(required=True)
|
|
comic = GenericReferenceField()
|
|
issues = ListField(ReferenceField(Issue))
|
|
|
|
def __str__(self):
|
|
s = "Volume(%s, %s, %s)" % (self.id, self.name, self.comic.title)
|
|
return s
|
|
|
|
|
|
class Comic(Document):
|
|
title = StringField(required=True)
|
|
publisher = ReferenceField(Publisher)
|
|
current_order = BooleanField()
|
|
completed = BooleanField()
|
|
issues = ListField(ReferenceField(Issue))
|
|
stories = ListField(ReferenceField(StoryArc))
|
|
|
|
def __str__(self):
|
|
if self.publisher is None:
|
|
s = "Comic(%s, %s, %s, %s)" % (self.title, self.publisher, self.current_order, self.completed)
|
|
return s
|
|
else:
|
|
s = "Comic(%s, %s, %s, %s)" % (
|
|
self.title, self.publisher.name, self.current_order, self.completed)
|
|
return s
|
|
|
|
|
|
class TradePaperback(Document):
|
|
comic = ReferenceField(Comic)
|
|
issue_start = StringField()
|
|
issue_end = StringField()
|
|
|
|
def __str__(self):
|
|
s = "TPB(%s)" % self.comic.title
|
|
return s
|
|
|
|
|
|
def get_publisher(name):
|
|
publisher = Publisher.objects(name=name)
|
|
if publisher.count() > 0:
|
|
return publisher[0]
|
|
else:
|
|
return None
|
|
|
|
|
|
def get_comic(title):
|
|
comic = Comic.objects(title=title)
|
|
if comic.count() > 0:
|
|
return comic[0]
|
|
else:
|
|
return None
|
|
|
|
|
|
def get_issue(title, number):
|
|
comic = get_comic(title)
|
|
issues = Issue.objects(number=number, comic=comic)
|
|
if issues.count() > 0:
|
|
return issues[0]
|
|
else:
|
|
return None
|
|
|
|
if __name__ == '__main__':
|
|
connect('comics')
|
|
for publisher in Publisher.objects:
|
|
pprint.pprint(publisher)
|
|
for artist in Artist.objects:
|
|
pprint.pprint(artist)
|
|
for comic in Comic.objects:
|
|
pprint.pprint(comic)
|
|
for issue in Issue.objects:
|
|
pprint.pprint(issue)
|
|
for story in StoryArc.objects:
|
|
pprint.pprint(story)
|
|
for tpb in TradePaperback.objects:
|
|
pprint.pprint(tpb)
|