You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
1.3 KiB
55 lines
1.3 KiB
import attr
|
|
import uuid
|
|
import pendulum
|
|
|
|
from datetime import datetime
|
|
|
|
def convert_date(text):
|
|
"""
|
|
Tries to parse a date into a DateTime instance
|
|
Returns `None` if it cannot be parsed
|
|
"""
|
|
try:
|
|
return pendulum.from_format(text, "DD/M/YYYY")
|
|
except ValueError:
|
|
try:
|
|
return (pendulum.from_format(text, "DD MMM")
|
|
.set(year=pendulum.now().year))
|
|
except ValueError:
|
|
return None
|
|
|
|
# Data type definitions of posts and comments
|
|
@attr.s
|
|
class Post:
|
|
content = attr.ib()
|
|
comments = attr.ib(factory=list)
|
|
date = attr.ib(factory=pendulum.now)
|
|
name = attr.ib(factory=lambda: uuid.uuid4().hex)
|
|
|
|
@attr.s
|
|
class Comment:
|
|
commenter = attr.ib()
|
|
content = attr.ib()
|
|
date = attr.ib(factory=pendulum.now)
|
|
name = attr.ib(factory=lambda: uuid.uuid4().hex)
|
|
|
|
@attr.s
|
|
class Conversation:
|
|
url = attr.ib()
|
|
name = attr.ib()
|
|
date : datetime = attr.ib(converter=convert_date)
|
|
messages = attr.ib(factory=list)
|
|
image_links = attr.ib(factory=list)
|
|
|
|
@attr.s
|
|
class Message:
|
|
name = attr.ib()
|
|
content = attr.ib()
|
|
|
|
# Remove the last 3 digits from FB's dates. They are not standard.
|
|
date : datetime = attr.ib(converter=lambda t: pendulum.from_timestamp(int(str(t)[0:-3])))
|
|
|
|
@attr.s
|
|
class Page:
|
|
name = attr.ib()
|
|
date = attr.ib(factory=pendulum.now)
|
|
|