mirror of
https://github.com/bookwyrm-social/bookwyrm.git
synced 2024-05-19 08:48:19 +00:00
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
""" handle reading a csv from librarything """
|
|
import csv
|
|
import re
|
|
import math
|
|
|
|
from bookwyrm import models
|
|
from bookwyrm.models import ImportItem
|
|
from bookwyrm.importer import Importer
|
|
|
|
|
|
class LibrarythingImporter(Importer):
|
|
service = "LibraryThing"
|
|
delimiter = "\t"
|
|
encoding = "ISO-8859-1"
|
|
# mandatory_fields : fields matching the book title and author
|
|
mandatory_fields = ["Title", "Primary Author"]
|
|
|
|
def parse_fields(self, initial):
|
|
data = {}
|
|
data["import_source"] = self.service
|
|
data["Book Id"] = initial["Book Id"]
|
|
data["Title"] = initial["Title"]
|
|
data["Author"] = initial["Primary Author"]
|
|
data["ISBN13"] = initial["ISBN"]
|
|
data["My Review"] = initial["Review"]
|
|
if initial["Rating"]:
|
|
data["My Rating"] = math.ceil(float(initial["Rating"]))
|
|
else:
|
|
data["My Rating"] = ""
|
|
data["Date Added"] = re.sub("\[|\]", "", initial["Entry Date"])
|
|
data["Date Started"] = re.sub("\[|\]", "", initial["Date Started"])
|
|
data["Date Read"] = re.sub("\[|\]", "", initial["Date Read"])
|
|
|
|
data["Exclusive Shelf"] = None
|
|
if data["Date Read"]:
|
|
data["Exclusive Shelf"] = "read"
|
|
elif data["Date Started"]:
|
|
data["Exclusive Shelf"] = "reading"
|
|
else:
|
|
data["Exclusive Shelf"] = "to-read"
|
|
|
|
return data
|