Esempio n. 1
0
 def save(self):
   """Warning: the source will be saved as well as the related objects
   (no commit options).
   Returns the source.
   """
   form_title = self.cleaned_data['title']
   form_username = self.cleaned_data['username']
   source_url = TwitterTimeline.SOURCE_URL
   source_name = "Twitter"
   source_pub_date = datetime.now(timezone.utc)
   provider = GeneratedFeed.TWITTER
   # Make sure that this specific user profile has a
   # corresponding user info to be sure it doesn't share
   # credentials with another's user profile just
   # by writting down this other's user's twitter username !
   same_twitter_info = self.user.userprofile.twitter_info
   same_twitter = TwitterTimeline.objects.filter(
     username=form_username).all()
   # url are unique for sources
   if same_twitter_info and same_twitter:
     return same_twitter[0].generated_feed.source
   if not same_twitter_info:
     new_twitter_user = TwitterUserInfo(username=form_username)
     new_twitter_user.save()
     self.user.userprofile.twitter_info = new_twitter_user
     self.user.userprofile.save()
     same_twitter_info = new_twitter_user
   any_twitter_sources = Reference.objects.filter(
     url = source_url,
     ).all()
   with transaction.atomic():
     if any_twitter_sources:
       twitter_source = any_twitter_sources[0]
     else:
       twitter_source = Reference(
         url=source_url, title=source_name,
         pub_date=source_pub_date)
     twitter_source.add_pin()
     twitter_source.save()
   with transaction.atomic():
     new_feed = GeneratedFeed(
       provider=provider,
       source=twitter_source, title=form_title)
     new_feed.last_update_check = (
       datetime
       .utcfromtimestamp(0)
       .replace(tzinfo=timezone.utc)
       )
     new_feed.save()
   with transaction.atomic():
     new_twitter = TwitterTimeline(
       username=form_username,
       generated_feed=new_feed,
       twitter_user_access_info = same_twitter_info)
     new_twitter.save()
     if twitter_source not in self.user.userprofile.sources.all():
       self.user.userprofile.sources.add(twitter_source)
     self.user.userprofile.generated_feeds.add(new_feed)
     self.user.userprofile.save()
   return twitter_source
Esempio n. 2
0
 def save(self):
   """Warning: the source will be saved as well as the related objects
   (no commit options).
   Returns the source.
   """
   form_url,_ = sanitize_url(self.cleaned_data["url"])
   form_title = self.cleaned_data["title"]
   form_feed_url,_ = sanitize_url(self.cleaned_data["feed_url"])
   if self.user.userprofile.web_feeds.filter(source__url=form_url).exists():
     # nothing to do
     return
   # try a bigger look-up anyway
   same_sources = WebFeed.objects.filter(source__url=form_url).all()
   # url are unique for sources
   if same_sources:
     new_feed = same_sources[0]
   else:
     if form_title:
       source_title = form_title
     else:
       source_title = build_reference_title_from_url(form_url)
     try:
       source_ref = Reference.objects.get(url=form_url)
     except ObjectDoesNotExist:
       source_ref = Reference(url=form_url,title=source_title,
                              pub_date=datetime.now(timezone.utc))
       source_ref.save()
     new_feed = WebFeed(source=source_ref)
     # assume that either form_feed_url or form_url have been
     # validated as a valid feed url
     new_feed.xmlURL = form_feed_url or form_url
     new_feed.last_update_check = datetime.utcfromtimestamp(0)\
                                          .replace(tzinfo=timezone.utc)
     new_feed.save()
   with transaction.atomic():
     source_ref.add_pin()
     source_ref.save()
     self.user.userprofile.sources.add(source_ref)
     self.user.userprofile.public_sources.add(source_ref)
     self.user.userprofile.web_feeds.add(new_feed)
     self.user.userprofile.save()
   return new_feed
Esempio n. 3
0
 def save(self):
   """Warning: the bookmark will be saved as well as the related objects
   (no commit options).
   Returns the bookmark.
   """
   url,_ = sanitize_url(self.cleaned_data["url"])
   title = self.cleaned_data["title"] \
           or build_reference_title_from_url(url)
   comment = self.cleaned_data["comment"]
   pub_date = self.cleaned_data["pub_date"] \
              or datetime.now(timezone.utc)
   src_url,_ = sanitize_url(self.cleaned_data["source_url"] \
                            or build_source_url_from_reference_url(url))
   src_title = self.cleaned_data["source_title"] \
              or build_reference_title_from_url(src_url)
   # Find or create a matching reference
   try:
     bookmarked_ref = Reference.objects.get(url=url)
     # Arbitrarily chose one of the possible sources
     src_query = bookmarked_ref.sources
     if src_query.count() > 1:
       ref_src = src_query.all()[0]
     else:
       ref_src = src_query.get()
   except ObjectDoesNotExist:
     try:
       ref_src = Reference.objects.get(url=src_url)
     except ObjectDoesNotExist:
       ref_src = Reference(url=src_url,title=src_title,pub_date=pub_date)
       ref_src.save()
     if src_url == url:
       bookmarked_ref = ref_src
     else:
       bookmarked_ref = Reference(url=url,
                                  title=title,
                                  pub_date=pub_date)
       bookmarked_ref.save()
       bookmarked_ref.sources.add(ref_src)
       bookmarked_ref.save()
   with transaction.atomic():
     try:
       bmk = UserBookmark.objects.get(owner=self.user,reference=bookmarked_ref)
     except ObjectDoesNotExist:
       bmk = UserBookmark(owner=self.user,reference=bookmarked_ref,
                          saved_date=datetime.now(timezone.utc))
       bookmarked_ref.add_pin()
       bmk.save()
       bookmarked_ref.save()
     # allow the user-specific comment to be changed and also prefix
     # it with the user specified title if it differs from the
     # existing reference title.
     if comment:
       new_comment = comment
     else:
       new_comment = bmk.comment
     if self.cleaned_data["title"] and title!=bookmarked_ref.title \
        and not new_comment.startswith(title):
       new_comment = "%s: %s" % (title,new_comment)
     if new_comment!=bmk.comment:
       bmk.comment = new_comment
       bmk.save()
   with transaction.atomic():
     if ref_src not in self.user.userprofile.sources.all():
       self.user.userprofile.sources.add(ref_src)
       self.user.userprofile.save()
   with transaction.atomic():
     for rust in ReferenceUserStatus\
       .objects.filter(owner=self.user,
                       reference=bookmarked_ref).all():
       rust.has_been_saved = True
       rust.save()
   return bmk