''' Handle user activity ''' from django.db import IntegrityError from fedireads import models from fedireads.books_manager import get_or_create_book from fedireads.sanitize_html import InputHtmlParser def create_review_from_activity(author, activity): ''' parse an activity json blob into a status ''' book_id = activity['inReplyToBook'] book_id = book_id.split('/')[-1] name = activity.get('name') rating = activity.get('rating') content = activity.get('content') published = activity.get('published') remote_id = activity['id'] book = get_or_create_book(book_id) review = create_review(author, book, name, content, rating) review.published_date = published review.remote_id = remote_id review.save() return review def create_review(user, book, name, content, rating): ''' a book review has been added ''' name = sanitize(name) content = sanitize(content) # no ratings outside of 0-5 if rating: rating = rating if 1 <= rating <= 5 else None else: rating = None return models.Review.objects.create( user=user, book=book, name=name, rating=rating, content=content, ) def create_comment_from_activity(author, activity): ''' parse an activity json blob into a status ''' book = activity['inReplyToBook'] book = book.split('/')[-1] content = activity.get('content') published = activity.get('published') remote_id = activity['id'] comment = create_comment(author, book, content) comment.published_date = published comment.remote_id = remote_id comment.save() return comment def create_comment(user, possible_book, content): ''' a book comment has been added ''' # throws a value error if the book is not found book = get_or_create_book(possible_book) content = sanitize(content) return models.Comment.objects.create( user=user, book=book, content=content, ) def create_status_from_activity(author, activity): ''' parse a status object out of an activity json blob ''' content = activity.get('content') reply_parent_id = activity.get('inReplyTo') reply_parent = get_status(reply_parent_id) remote_id = activity['id'] if models.Status.objects.filter(remote_id=remote_id).count(): return None status = create_status(author, content, reply_parent=reply_parent, remote_id=remote_id) status.published_date = activity.get('published') status.save() return status def create_favorite_from_activity(user, activity): ''' create a new favorite entry ''' status = get_status(activity['object']) remote_id = activity['id'] try: return models.Favorite.objects.create( status=status, user=user, remote_id=remote_id, ) except IntegrityError: return models.Favorite.objects.get(status=status, user=user) def create_boost_from_activity(user, activity): ''' create a new boost activity ''' status = get_status(activity['object']) remote_id = activity['id'] try: return models.Boost.objects.create( status=status, user=user, remote_id=remote_id, ) except IntegrityError: return models.Boost.objects.get(status=status, user=user) def get_status(absolute_id): ''' find a status in the database ''' return get_by_absolute_id(absolute_id, models.Status) def get_favorite(absolute_id): ''' find a status in the database ''' return get_by_absolute_id(absolute_id, models.Favorite) def get_by_absolute_id(absolute_id, model): ''' generalized function to get from a model with a remote_id field ''' if not absolute_id: return None # check if it's a remote status try: return model.objects.get(remote_id=absolute_id) except model.DoesNotExist: pass # try finding a local status with that id local_id = absolute_id.split('/')[-1] try: if hasattr(model.objects, 'select_subclasses'): possible_match = model.objects.select_subclasses().get(id=local_id) else: possible_match = model.objects.get(id=local_id) except model.DoesNotExist: return None # make sure it's not actually a remote status with an id that # clashes with a local id if possible_match.absolute_id == absolute_id: return possible_match return None def create_status(user, content, reply_parent=None, mention_books=None, remote_id=None): ''' a status update ''' # TODO: handle @'ing users # sanitize input html parser = InputHtmlParser() parser.feed(content) content = parser.get_output() status = models.Status.objects.create( user=user, content=content, reply_parent=reply_parent, remote_id=remote_id, ) if mention_books: for book in mention_books: status.mention_books.add(book) return status def create_tag(user, possible_book, name): ''' add a tag to a book ''' book = get_or_create_book(possible_book) try: tag = models.Tag.objects.create(name=name, book=book, user=user) except IntegrityError: return models.Tag.objects.get(name=name, book=book, user=user) return tag def create_notification(user, notification_type, related_user=None, \ related_book=None, related_status=None): ''' let a user know when someone interacts with their content ''' if user == related_user: # don't create notification when you interact with your own stuff return models.Notification.objects.create( user=user, related_book=related_book, related_user=related_user, related_status=related_status, notification_type=notification_type, ) def sanitize(content): ''' remove invalid html from free text ''' parser = InputHtmlParser() parser.feed(content) return parser.get_output()