def test_delete(self):
series, created = Series.from_dict({"name": "Secret Files"})
self.assertTrue(created)
self.assertIsNotNone(series.id)
deleted = series.delete()
self.assertIsNone(series.id)
self.assertEquals((1, {"series.Series": 1}), deleted)
series, created = Series.from_dict({
"name":
"Secret Files",
"links": [{
"url": "https://secrectfiles.com"
}]
})
self.assertTrue(created)
self.assertIsNotNone(series.id)
deleted = series.delete()
self.assertIsNone(series.id)
self.assertEquals(
(3, {
"series.Series": 1,
"series.Series_links": 1,
"links.Link": 1
}),
deleted,
)
def from_dict(cls: Type[T], data: Dict) -> Tuple[T, bool]:
"""Create from dict.
Returns True if was crated, i. e. was not found in the DB.
"""
defaults: Dict = {}
if "series" in data and data["series"]:
defaults["series"] = Series.from_dict(data["series"])[0]
if "volume" in data and data["volume"]:
defaults["volume"] = data["volume"]
book, created = Book.objects.get_or_create(title=data["title"],
defaults=defaults)
if "authors" in data and data["authors"]:
for i in data["authors"]:
book.authors.add(Person.from_dict(i)[0])
if "genres" in data and data["genres"]:
for i in data["genres"]:
book.genres.add(Genre.from_dict(i)[0])
if "links" in data and data["links"]:
for i in data["links"]:
book.links.add(Link.from_dict(i)[0])
return book, created
def test_search(self):
series, created = Series.from_dict({"name": "Secret Files"})
self.assertTrue(created)
self.assertIsNotNone(series.id)
series, created = Series.from_dict({"name": "Ran"})
self.assertTrue(created)
self.assertIsNotNone(series.id)
series, created = Series.from_dict({"name": "Random Files"})
self.assertTrue(created)
self.assertIsNotNone(series.id)
self.assertEquals(3, Series.objects.all().count())
self.assertEquals(2, Series.search("files").count())
self.assertEquals(2, Series.search("ran").count())
def test_from_to_dict(self):
link, created = Link.objects.get_or_create(
link="https://secrectfiles.com")
self.assertTrue(created)
self.assertIsNotNone(link.id)
series, created = Series.objects.get_or_create(name="Secret Files")
series.links.add(link)
self.assertTrue(created)
self.assertIsNotNone(series.id)
self.assertEquals(
{
"name": "Secret Files",
"links": [{
"url": "https://secrectfiles.com"
}]
},
series.to_dict(),
)
self.assertEquals(
(series, False),
Series.from_dict({
"name": "Secret Files",
"links": [{
"url": "https://secrectfiles.com"
}]
}),
)
self.assertEquals((series, False),
Series.from_dict({"name": "Secret Files"}))
series, created = Series.objects.get_or_create(name="Random")
self.assertTrue(created)
self.assertIsNotNone(series.id)
self.assertEquals({"name": "Random", "links": None}, series.to_dict())
self.assertEquals((series, False),
Series.from_dict({
"name": "Random",
"links": None
}))
self.assertEquals((series, False), Series.from_dict({"name":
"Random"}))
def test_get(self):
series, created = Series.from_dict({"name": "Secret Files"})
self.assertTrue(created)
self.assertIsNotNone(series.id)
series2 = Series.get("Secret Files")
self.assertIsNotNone(series2)
self.assertEquals(series, series2)
series2 = Series.get("files")
self.assertIsNotNone(series2)
self.assertEquals(series, series2)
series2 = Series.get(str(series.id))
self.assertIsNotNone(series2)
self.assertEquals(series, series2)
def test_get_or_create(self):
series, created = Series.from_dict({"name": "Secret Files"})
self.assertTrue(created)
self.assertIsNotNone(series.id)
self.assertEquals(1, Series.objects.count())
series2 = Series.get_or_create("Secret Files")
self.assertIsNotNone(series2)
self.assertEquals(series, series2)
self.assertEquals(1, Series.objects.count())
series2 = Series.get_or_create(str(series.id))
self.assertIsNotNone(series2)
self.assertEquals(series, series2)
self.assertEquals(1, Series.objects.count())
series2 = Series.get_or_create("Secret Papers")
self.assertIsNotNone(series2)
self.assertNotEquals(series, series2)
self.assertEquals(2, Series.objects.count())
def test_print(self):
series, created = Series.from_dict({"name": "Secret Files"})
self.assertTrue(created)
self.assertIsNotNone(series.id)
with StringIO() as cout:
series.print(cout)
self.assertEquals(
"Field Value "
+
" \n=================================="
+
"==================================================================\n"
+
"Id 1 "
+
" \n__________________________________"
+
"__________________________________________________________________\n"
+
"Name Secret Files "
+
" \n__________________________________"
+
"__________________________________________________________________\n"
+
"Links "
+
" \n__________________________________"
+
"__________________________________________________________________\n"
+
"Books "
+
" \n__________________________________"
+
"__________________________________________________________________\n",
cout.getvalue(),
)
def test_from_to_dict(self):
book, created = Book.objects.get_or_create(title="Cool")
self.assertTrue(created)
self.assertIsNotNone(book.id)
self.assertEquals(
{
"title": "Cool",
"authors": None,
"series": None,
"volume": 0,
"genres": None,
"links": None,
},
book.to_dict(),
)
self.assertEquals((book, False), Book.from_dict(book.to_dict()))
self.assertEquals((book, False), Book.from_dict({"title": "Cool"}))
author, created = Person.from_dict({"name": "Max Mustermann"})
self.assertTrue(created)
self.assertIsNotNone(author.id)
series, created = Series.from_dict({"name": "Secret Files"})
self.assertTrue(created)
self.assertIsNotNone(series.id)
genre, created = Genre.from_dict({"name": "Fiction"})
self.assertTrue(created)
self.assertIsNotNone(genre.id)
link, created = Link.from_dict({"url": "https://example.com"})
self.assertTrue(created)
self.assertIsNotNone(link.id)
book, created = Book.objects.get_or_create(title="Example",
series=series,
volume=1)
book.authors.add(author)
book.genres.add(genre)
book.links.add(link)
self.assertTrue(created)
self.assertIsNotNone(book.id)
self.assertEquals(
{
"title": "Example",
"authors": [{
"name": "Max Mustermann",
"links": None
}],
"series": {
"name": "Secret Files",
"links": None
},
"volume": 1,
"genres": [{
"name": "Fiction"
}],
"links": [{
"url": "https://example.com"
}],
},
book.to_dict(),
)
self.assertEquals((book, False), Book.from_dict(book.to_dict()))
def test_edit(self):
series, created = Series.from_dict({"name": "Test Series"})
self.assertTrue(created)
self.assertIsNotNone(series.id)
person, created = Person.from_dict({"name": "John Do"})
self.assertTrue(created)
self.assertIsNotNone(person.id)
book, created = Book.from_dict({
"title": "Test2 Book",
"authors": [person.to_dict()],
"series": series.to_dict(),
"volume": 1.0,
"genres": [{
"name": "Romance"
}],
})
self.assertTrue(created)
self.assertIsNotNone(book.id)
self.assertEquals(series, book.series)
self.assertEquals(person, book.authors.first())
book.edit("title", "IEEE Test Book")
self.assertEquals("IEEE Test Book", book.title)
book.edit("author", "Jane Do")
self.assertEquals(2, book.authors.count())
self.assertEquals("Jane Do", str(book.authors.all()[0]))
self.assertEquals("John Do", str(book.authors.all()[1]))
book.edit("author", str(person.id))
self.assertEquals(1, book.authors.count())
self.assertEquals("Jane Do", str(book.authors.all()[0]))
series, created = Series.from_dict({"name": "Space Series"})
self.assertTrue(created)
self.assertIsNotNone(series.id)
book.edit("series", str(series.id))
self.assertEquals(series, book.series)
book.edit("volume", 0.75)
self.assertEquals(0.75, book.volume)
series, created = Series.from_dict({"name": "Deep Space Series"})
self.assertTrue(created)
self.assertIsNotNone(series.id)
book.edit("series", "Deep Space")
self.assertEquals(series, book.series)
book.edit("genre", "SciFi")
self.assertEquals(2, book.genres.count())
self.assertEquals("Romance", book.genres.first().name)
self.assertEquals("SciFi", book.genres.last().name)
book.edit("genre", "1")
self.assertEquals(1, book.genres.count())
self.assertEquals("SciFi", book.genres.first().name)
book.edit("link", "https://deep.space")
self.assertEquals(1, book.links.count())
self.assertEquals("https://deep.space", book.links.first().link)
book.edit("link", "https://janedo.com/test2book")
self.assertEquals(2, book.links.count())
self.assertEquals("https://janedo.com/test2book",
book.links.last().link)
book.edit("link", "https://deep.space")
self.assertEquals(1, book.links.count())
self.assertEquals("https://janedo.com/test2book",
book.links.first().link)
def _series(args: Namespace, file: TextIO = sys.stdout):
series: Optional[Series] = None
if args.subparser == "add":
series, created = Series.from_dict({
"name":
args.name,
"links":
[Link.get_or_create(link).to_dict() for link in args.link],
})
if created:
stdout.write(
_('Successfully added series "%(name)s" with id "%(pk)d".') % {
"name": series.name,
"pk": series.pk
},
"=",
file=file,
)
series.print(file)
else:
stdout.write(
_('The series "%(name)s" already exists with id "%(pk)d", aborting...'
) % {
"name": series.name,
"pk": series.pk
},
"",
file=file,
)
elif args.subparser == "delete":
series = Series.get(args.series)
if series:
series.delete()
stdout.write(
_('Successfully deleted series with id "%(pk)d".') %
{"pk": series.pk},
"",
file=file,
)
else:
stdout.write(_("No series found."), "", file=file)
elif args.subparser == "edit":
series = Series.get(args.series)
if series:
series.edit(args.field, args.value)
stdout.write(
_('Successfully edited series "%(name)s" with id "%(pk)d".') %
{
"name": series.name,
"pk": series.pk
},
"",
file=file,
)
series.print(file)
else:
stdout.write(_("No series found."), "", file=file)
elif args.subparser == "info":
series = Series.get(args.series)
if series:
series.print(file)
else:
stdout.write(_("No series found."), "", file=file)
elif args.subparser == "list":
if args.search:
series = Series.search(args.search)
else:
series = Series.objects.all()
stdout.write(
[_("Id"), _("Name"), _("Number of books")],
"=", [0.05, 0.8],
file=file)
for i, has_next in lookahead(series):
stdout.write(
[i.id, i.name, i.books.count()],
"_" if has_next else "=",
[0.05, 0.8],
file=file,
)
