mirror of
https://github.com/samuelclay/NewsBlur.git
synced 2025-04-13 09:42:01 +00:00
925 lines
No EOL
36 KiB
Python
925 lines
No EOL
36 KiB
Python
import datetime
|
|
import time
|
|
import random
|
|
import re
|
|
from django.shortcuts import render_to_response, get_object_or_404
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.template import RequestContext
|
|
from django.db import IntegrityError
|
|
from django.views.decorators.cache import never_cache
|
|
from django.core.urlresolvers import reverse
|
|
from django.contrib.auth import login as login_user
|
|
from django.contrib.auth import logout as logout_user
|
|
from django.contrib.auth.models import User
|
|
from django.http import HttpResponse, HttpResponseRedirect, HttpResponseForbidden, Http404
|
|
from django.conf import settings
|
|
from django.core.mail import mail_admins
|
|
from collections import defaultdict
|
|
from operator import itemgetter
|
|
from mongoengine.queryset import OperationError
|
|
from apps.analyzer.models import MClassifierTitle, MClassifierAuthor, MClassifierFeed, MClassifierTag
|
|
from apps.analyzer.models import apply_classifier_titles, apply_classifier_feeds, apply_classifier_authors, apply_classifier_tags
|
|
from apps.analyzer.models import get_classifiers_for_user
|
|
from apps.reader.models import UserSubscription, UserSubscriptionFolders, MUserStory, Feature
|
|
from apps.reader.forms import SignupForm, LoginForm, FeatureForm
|
|
try:
|
|
from apps.rss_feeds.models import Feed, MFeedPage, DuplicateFeed, MStory, MStarredStory, FeedLoadtime
|
|
except:
|
|
pass
|
|
from utils import json_functions as json
|
|
from utils.user_functions import get_user, ajax_login_required
|
|
from utils.feed_functions import relative_timesince
|
|
from utils.feed_functions import add_object_to_folder
|
|
from utils.story_functions import format_story_link_date__short
|
|
from utils.story_functions import format_story_link_date__long
|
|
from utils.story_functions import bunch
|
|
from utils.story_functions import story_score
|
|
from utils import log as logging
|
|
from utils.timezones.utilities import localtime_for_timezone
|
|
|
|
SINGLE_DAY = 60*60*24
|
|
|
|
@never_cache
|
|
def index(request):
|
|
if request.method == "POST":
|
|
if request.POST['submit'] == 'login':
|
|
login_form = LoginForm(request.POST, prefix='login')
|
|
signup_form = SignupForm(prefix='signup')
|
|
else:
|
|
login_form = LoginForm(prefix='login')
|
|
signup_form = SignupForm(request.POST, prefix='signup')
|
|
else:
|
|
login_form = LoginForm(prefix='login')
|
|
signup_form = SignupForm(prefix='signup')
|
|
|
|
user = get_user(request)
|
|
features = Feature.objects.all()[:3]
|
|
feature_form = None
|
|
if request.user.is_staff:
|
|
feature_form = FeatureForm()
|
|
|
|
feed_count = 0
|
|
active_count = 0
|
|
train_count = 0
|
|
if request.user.is_authenticated():
|
|
feed_count = UserSubscription.objects.filter(user=request.user).count()
|
|
active_count = UserSubscription.objects.filter(user=request.user, active=True).count()
|
|
train_count = UserSubscription.objects.filter(user=request.user, active=True, is_trained=False, feed__stories_last_month__gte=1).count()
|
|
|
|
howitworks_page = random.randint(0, 5)
|
|
return render_to_response('reader/feeds.xhtml', {
|
|
'user_profile': user.profile,
|
|
'login_form': login_form,
|
|
'signup_form': signup_form,
|
|
'feature_form': feature_form,
|
|
'features': features,
|
|
'start_import_from_google_reader': request.session.get('import_from_google_reader', False),
|
|
'howitworks_page': howitworks_page,
|
|
'feed_count': feed_count,
|
|
'active_count': active_count,
|
|
'train_count': active_count - train_count,
|
|
'account_images': range(1, 4),
|
|
}, context_instance=RequestContext(request))
|
|
|
|
@never_cache
|
|
def login(request):
|
|
code = -1
|
|
if request.method == "POST":
|
|
form = LoginForm(request.POST, prefix='login')
|
|
if form.is_valid():
|
|
login_user(request, form.get_user())
|
|
if request.POST.get('api'):
|
|
logging.info(" ---> [%s] ~FG~BB~SKiPhone Login~FW" % form.get_user())
|
|
code = 1
|
|
else:
|
|
logging.info(" ---> [%s] ~FG~BBLogin~FW" % form.get_user())
|
|
return HttpResponseRedirect(reverse('index'))
|
|
|
|
if request.POST.get('api'):
|
|
return HttpResponse(json.encode(dict(code=code)), mimetype='application/json')
|
|
else:
|
|
return index(request)
|
|
|
|
@never_cache
|
|
def signup(request):
|
|
if request.method == "POST":
|
|
form = SignupForm(prefix='signup', data=request.POST)
|
|
if form.is_valid():
|
|
new_user = form.save()
|
|
login_user(request, new_user)
|
|
logging.info(" ---> [%s] ~FG~SB~BBNEW SIGNUP~FW" % new_user)
|
|
return HttpResponseRedirect(reverse('index'))
|
|
|
|
return index(request)
|
|
|
|
@never_cache
|
|
def logout(request):
|
|
logging.info(" ---> [%s] ~FG~BBLogout~FW" % request.user)
|
|
logout_user(request)
|
|
|
|
if request.GET.get('api'):
|
|
return HttpResponse(json.encode(dict(code=1)), mimetype='application/json')
|
|
else:
|
|
return HttpResponseRedirect(reverse('index'))
|
|
|
|
@json.json_view
|
|
def load_feeds(request):
|
|
user = get_user(request)
|
|
feeds = {}
|
|
not_yet_fetched = False
|
|
|
|
try:
|
|
folders = UserSubscriptionFolders.objects.get(user=user)
|
|
except UserSubscriptionFolders.DoesNotExist:
|
|
data = dict(feeds=[], folders=[])
|
|
return data
|
|
except UserSubscriptionFolders.MultipleObjectsReturned:
|
|
UserSubscriptionFolders.objects.filter(user=user)[1:].delete()
|
|
folders = UserSubscriptionFolders.objects.get(user=user)
|
|
|
|
user_subs = UserSubscription.objects.select_related('feed', 'feed__feed_icon').filter(user=user)
|
|
|
|
for sub in user_subs:
|
|
feeds[sub.feed.pk] = sub.canonical()
|
|
if feeds[sub.feed.pk].get('not_yet_fetched'):
|
|
not_yet_fetched = True
|
|
if not sub.feed.active and not sub.feed.has_feed_exception and not sub.feed.has_page_exception:
|
|
sub.feed.count_subscribers()
|
|
sub.feed.schedule_feed_fetch_immediately()
|
|
|
|
if not_yet_fetched:
|
|
for f in feeds:
|
|
if 'not_yet_fetched' not in feeds[f]:
|
|
feeds[f]['not_yet_fetched'] = False
|
|
|
|
starred_count = MStarredStory.objects(user_id=user.pk).count()
|
|
|
|
data = {
|
|
'feeds': feeds,
|
|
'folders': json.decode(folders.folders),
|
|
'starred_count': starred_count,
|
|
}
|
|
return data
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def load_feeds_iphone(request):
|
|
user = get_user(request)
|
|
feeds = {}
|
|
|
|
try:
|
|
folders = UserSubscriptionFolders.objects.get(user=user)
|
|
except UserSubscriptionFolders.DoesNotExist:
|
|
data = dict(folders=[])
|
|
return data
|
|
|
|
user_subs = UserSubscription.objects.select_related('feed').filter(user=user)
|
|
|
|
for sub in user_subs:
|
|
if sub.needs_unread_recalc:
|
|
sub.calculate_feed_scores(silent=True)
|
|
feeds[sub.feed.pk] = {
|
|
'id': sub.feed.pk,
|
|
'feed_title': sub.user_title or sub.feed.feed_title,
|
|
'feed_link': sub.feed.feed_link,
|
|
'ps': sub.unread_count_positive,
|
|
'nt': sub.unread_count_neutral,
|
|
'ng': sub.unread_count_negative,
|
|
}
|
|
|
|
folders = json.decode(folders.folders)
|
|
flat_folders = {}
|
|
|
|
def make_feeds_folder(items, parent_folder="", depth=0):
|
|
for item in items:
|
|
if isinstance(item, int) and item in feeds:
|
|
feed = feeds[item]
|
|
if not parent_folder:
|
|
parent_folder = ' '
|
|
if parent_folder in flat_folders:
|
|
flat_folders[parent_folder].append(feed)
|
|
else:
|
|
flat_folders[parent_folder] = [feed]
|
|
elif isinstance(item, dict):
|
|
for folder_name in item:
|
|
folder = item[folder_name]
|
|
flat_folder_name = "%s%s%s" % (
|
|
parent_folder,
|
|
" - " if parent_folder else "",
|
|
folder_name
|
|
)
|
|
make_feeds_folder(folder, flat_folder_name, depth+1)
|
|
|
|
make_feeds_folder(folders)
|
|
data = dict(flat_folders=flat_folders, user=user.username)
|
|
return data
|
|
|
|
@json.json_view
|
|
def refresh_feeds(request):
|
|
user = get_user(request)
|
|
feeds = {}
|
|
user_subs = UserSubscription.objects.select_related('feed').filter(user=user, active=True)
|
|
UNREAD_CUTOFF = datetime.datetime.utcnow() - datetime.timedelta(days=settings.DAYS_OF_UNREAD)
|
|
favicons_fetching = [int(f) for f in request.POST.getlist('favicons_fetching')]
|
|
|
|
for sub in user_subs:
|
|
if (sub.needs_unread_recalc or
|
|
sub.unread_count_updated < UNREAD_CUTOFF or
|
|
sub.oldest_unread_story_date < UNREAD_CUTOFF):
|
|
sub.calculate_feed_scores(silent=True)
|
|
feeds[sub.feed.pk] = {
|
|
'ps': sub.unread_count_positive,
|
|
'nt': sub.unread_count_neutral,
|
|
'ng': sub.unread_count_negative,
|
|
}
|
|
if sub.feed.has_feed_exception or sub.feed.has_page_exception:
|
|
feeds[sub.feed.pk]['has_exception'] = True
|
|
feeds[sub.feed.pk]['exception_type'] = 'feed' if sub.feed.has_feed_exception else 'page'
|
|
feeds[sub.feed.pk]['feed_address'] = sub.feed.feed_address
|
|
feeds[sub.feed.pk]['exception_code'] = sub.feed.exception_code
|
|
if request.POST.get('check_fetch_status', False):
|
|
feeds[sub.feed.pk]['not_yet_fetched'] = not sub.feed.fetched_once
|
|
if sub.feed.pk in favicons_fetching:
|
|
feeds[sub.feed.pk]['favicon'] = sub.feed.icon.data
|
|
feeds[sub.feed.pk]['favicon_color'] = sub.feed.icon.color
|
|
feeds[sub.feed.pk]['favicon_fetching'] = bool(not (sub.feed.icon.not_found or sub.feed.icon.data))
|
|
|
|
return {'feeds': feeds}
|
|
|
|
@json.json_view
|
|
def load_single_feed(request):
|
|
user = get_user(request)
|
|
offset = int(request.REQUEST.get('offset', 0))
|
|
limit = int(request.REQUEST.get('limit', 30))
|
|
page = int(request.REQUEST.get('page', 0))
|
|
if page:
|
|
offset = limit * page
|
|
feed_id = None
|
|
try:
|
|
feed_id = int(request.REQUEST.get('feed_id', 0))
|
|
except ValueError:
|
|
feed_id_matches = re.search(r'(\d+)', request.REQUEST['feed_id'])
|
|
if feed_id_matches: feed_id = int(feed_id_matches.group(1))
|
|
dupe_feed_id = None
|
|
if not feed_id:
|
|
raise Http404
|
|
|
|
try:
|
|
feed = Feed.objects.get(id=feed_id)
|
|
except Feed.DoesNotExist:
|
|
feed_address = request.REQUEST.get('feed_address')
|
|
dupe_feed = DuplicateFeed.objects.filter(duplicate_address=feed_address)
|
|
if dupe_feed:
|
|
feed = dupe_feed[0].feed
|
|
dupe_feed_id = feed_id
|
|
else:
|
|
raise Http404
|
|
|
|
force_update = request.GET.get('force_update', False)
|
|
|
|
start = datetime.datetime.utcnow()
|
|
stories = feed.get_stories(offset, limit)
|
|
|
|
if force_update:
|
|
feed.update(force_update)
|
|
|
|
# Get intelligence classifier for user
|
|
classifier_feeds = MClassifierFeed.objects(user_id=user.pk, feed_id=feed_id)
|
|
classifier_authors = MClassifierAuthor.objects(user_id=user.pk, feed_id=feed_id)
|
|
classifier_titles = MClassifierTitle.objects(user_id=user.pk, feed_id=feed_id)
|
|
classifier_tags = MClassifierTag.objects(user_id=user.pk, feed_id=feed_id)
|
|
|
|
usersub = UserSubscription.objects.get(user=user, feed=feed)
|
|
userstories = []
|
|
userstories_db = MUserStory.objects(user_id=user.pk,
|
|
feed_id=feed.pk,
|
|
read_date__gte=usersub.mark_read_date)
|
|
starred_stories = MStarredStory.objects(user_id=user.pk, story_feed_id=feed_id).only('story_guid', 'starred_date')
|
|
starred_stories = dict([(story.story_guid, story.starred_date) for story in starred_stories])
|
|
|
|
for us in userstories_db:
|
|
if hasattr(us.story, 'story_guid') and isinstance(us.story.story_guid, unicode):
|
|
userstories.append(us.story.story_guid)
|
|
elif hasattr(us.story, 'id') and isinstance(us.story.id, unicode):
|
|
userstories.append(us.story.id) # TODO: Remove me after migration from story.id->guid
|
|
|
|
for story in stories:
|
|
[x.rewind() for x in [classifier_feeds, classifier_authors, classifier_tags, classifier_titles]]
|
|
story_date = localtime_for_timezone(story['story_date'], user.profile.timezone)
|
|
now = localtime_for_timezone(datetime.datetime.now(), user.profile.timezone)
|
|
story['short_parsed_date'] = format_story_link_date__short(story_date, now)
|
|
story['long_parsed_date'] = format_story_link_date__long(story_date, now)
|
|
if story['id'] in userstories:
|
|
story['read_status'] = 1
|
|
elif not story.get('read_status') and story['story_date'] < usersub.mark_read_date:
|
|
story['read_status'] = 1
|
|
elif not story.get('read_status') and story['story_date'] > usersub.last_read_date:
|
|
story['read_status'] = 0
|
|
if story['id'] in starred_stories:
|
|
story['starred'] = True
|
|
starred_date = localtime_for_timezone(starred_stories[story['id']], user.profile.timezone)
|
|
story['starred_date'] = format_story_link_date__long(starred_date, now)
|
|
story['intelligence'] = {
|
|
'feed': apply_classifier_feeds(classifier_feeds, feed),
|
|
'author': apply_classifier_authors(classifier_authors, story),
|
|
'tags': apply_classifier_tags(classifier_tags, story),
|
|
'title': apply_classifier_titles(classifier_titles, story),
|
|
}
|
|
|
|
# Intelligence
|
|
feed_tags = json.decode(feed.data.popular_tags) if feed.data.popular_tags else []
|
|
feed_authors = json.decode(feed.data.popular_authors) if feed.data.popular_authors else []
|
|
classifiers = get_classifiers_for_user(user, feed_id, classifier_feeds,
|
|
classifier_authors, classifier_titles, classifier_tags)
|
|
|
|
usersub.feed_opens += 1
|
|
usersub.save()
|
|
|
|
diff = datetime.datetime.utcnow()-start
|
|
timediff = float("%s.%.2s" % (diff.seconds, (diff.microseconds / 1000)))
|
|
last_update = relative_timesince(feed.last_update)
|
|
logging.info(" ---> [%s] ~FYLoading feed: ~SB%s ~SN(%s seconds)" % (request.user, feed, timediff))
|
|
FeedLoadtime.objects.create(feed=feed, loadtime=timediff)
|
|
|
|
data = dict(stories=stories,
|
|
feed_tags=feed_tags,
|
|
feed_authors=feed_authors,
|
|
classifiers=classifiers,
|
|
last_update=last_update,
|
|
feed_id=feed.pk)
|
|
|
|
if dupe_feed_id: data['dupe_feed_id'] = dupe_feed_id
|
|
|
|
return data
|
|
|
|
def load_feed_page(request):
|
|
feed_id = None
|
|
try:
|
|
feed_id = int(request.REQUEST.get('feed_id', 0))
|
|
except ValueError:
|
|
feed_id_matches = re.search(r'(\d+)', request.REQUEST['feed_id'])
|
|
if feed_id_matches: feed_id = int(feed_id_matches.group(1))
|
|
if not feed_id:
|
|
raise Http404
|
|
|
|
data = MFeedPage.get_data(feed_id=feed_id)
|
|
|
|
if not data:
|
|
data = "Fetching feed..."
|
|
|
|
return HttpResponse(data, mimetype='text/html')
|
|
|
|
@json.json_view
|
|
def load_starred_stories(request):
|
|
user = get_user(request)
|
|
offset = int(request.REQUEST.get('offset', 0))
|
|
limit = int(request.REQUEST.get('limit', 10))
|
|
page = int(request.REQUEST.get('page', 0))
|
|
if page: offset = limit * page
|
|
|
|
mstories = MStarredStory.objects(user_id=user.pk).order_by('-starred_date')[offset:offset+limit]
|
|
stories = Feed.format_stories(mstories)
|
|
|
|
for story in stories:
|
|
story_date = localtime_for_timezone(story['story_date'], user.profile.timezone)
|
|
now = localtime_for_timezone(datetime.datetime.now(), user.profile.timezone)
|
|
story['short_parsed_date'] = format_story_link_date__short(story_date, now)
|
|
story['long_parsed_date'] = format_story_link_date__long(story_date, now)
|
|
starred_date = localtime_for_timezone(story['starred_date'], user.profile.timezone)
|
|
story['starred_date'] = format_story_link_date__long(starred_date, now)
|
|
story['read_status'] = 1
|
|
story['starred'] = True
|
|
story['intelligence'] = {
|
|
'feed': 0,
|
|
'author': 0,
|
|
'tags': 0,
|
|
'title': 0,
|
|
}
|
|
|
|
logging.info(" ---> [%s] ~FCLoading starred stories: ~SB%s stories" % (request.user, len(stories)))
|
|
|
|
return dict(stories=stories)
|
|
|
|
@json.json_view
|
|
def load_river_stories(request):
|
|
limit = 25
|
|
offset = 0
|
|
start = datetime.datetime.utcnow()
|
|
user = get_user(request)
|
|
feed_ids = [int(feed_id) for feed_id in request.POST.getlist('feeds') if feed_id]
|
|
original_feed_ids = list(feed_ids)
|
|
page = int(request.REQUEST.get('page', 0))+1
|
|
read_stories_count = int(request.REQUEST.get('read_stories_count', 0))
|
|
bottom_delta = datetime.timedelta(days=settings.DAYS_OF_UNREAD)
|
|
|
|
if not feed_ids:
|
|
logging.info(" ---> [%s] ~FCLoading empty river stories: page %s" % (
|
|
request.user, page))
|
|
return dict(stories=[])
|
|
|
|
# Fetch all stories at and before the page number.
|
|
# Not a single page, because reading stories can move them up in the unread order.
|
|
# `read_stories_count` is an optimization, works best when all 25 stories before have been read.
|
|
limit = limit * page - read_stories_count
|
|
|
|
# Read stories to exclude
|
|
read_stories = MUserStory.objects(user_id=user.pk, feed_id__in=feed_ids).only('story')
|
|
read_stories = [rs.story.id for rs in read_stories]
|
|
|
|
# Determine mark_as_read dates for all feeds to ignore all stories before this date.
|
|
# max_feed_count = 0
|
|
feed_counts = {}
|
|
feed_last_reads = {}
|
|
for feed_id in feed_ids:
|
|
try:
|
|
usersub = UserSubscription.objects.get(feed__pk=feed_id, user=user)
|
|
except UserSubscription.DoesNotExist:
|
|
continue
|
|
feed_counts[feed_id] = (usersub.unread_count_negative * 1+
|
|
usersub.unread_count_neutral * 10+
|
|
usersub.unread_count_positive * 20)
|
|
# if feed_counts[feed_id] > max_feed_count:
|
|
# max_feed_count = feed_counts[feed_id]
|
|
feed_last_reads[feed_id] = int(time.mktime(usersub.mark_read_date.timetuple()))
|
|
feed_counts = sorted(feed_counts.items(), key=itemgetter(1))[:25]
|
|
feed_ids = [f[0] for f in feed_counts]
|
|
feed_last_reads = dict([(str(feed_id), feed_last_reads[feed_id]) for feed_id in feed_ids])
|
|
feed_counts = dict(feed_counts)
|
|
|
|
# After excluding read stories, all that's left are stories
|
|
# past the mark_read_date. Everything returned is guaranteed to be unread.
|
|
mstories = MStory.objects(
|
|
id__nin=read_stories,
|
|
story_feed_id__in=feed_ids,
|
|
story_date__gte=start - bottom_delta
|
|
).map_reduce("""function() {
|
|
var d = feed_last_reads[this[~story_feed_id]];
|
|
if (this[~story_date].getTime()/1000 > d) {
|
|
emit(this[~id], this);
|
|
}
|
|
}""",
|
|
"""function(key, values) {
|
|
return values[0];
|
|
}""",
|
|
scope={
|
|
'feed_last_reads': feed_last_reads
|
|
}
|
|
)
|
|
mstories = [story.value for story in mstories]
|
|
|
|
mstories = sorted(mstories, cmp=lambda x, y: cmp(story_score(y, bottom_delta), story_score(x, bottom_delta)))
|
|
|
|
# story_feed_counts = defaultdict(int)
|
|
# mstories_pruned = []
|
|
# for story in mstories:
|
|
# print story['story_title'], story_feed_counts[story['story_feed_id']]
|
|
# if story_feed_counts[story['story_feed_id']] >= 3: continue
|
|
# mstories_pruned.append(story)
|
|
# story_feed_counts[story['story_feed_id']] += 1
|
|
stories = []
|
|
for i, story in enumerate(mstories):
|
|
if i < offset: continue
|
|
if i >= offset + limit: break
|
|
stories.append(bunch(story))
|
|
stories = Feed.format_stories(stories)
|
|
found_feed_ids = list(set([story['story_feed_id'] for story in stories]))
|
|
|
|
# Find starred stories
|
|
starred_stories = MStarredStory.objects(
|
|
user_id=user.pk,
|
|
story_feed_id__in=found_feed_ids
|
|
).only('story_guid', 'starred_date')
|
|
starred_stories = dict([(story.story_guid, story.starred_date)
|
|
for story in starred_stories])
|
|
|
|
# Intelligence classifiers for all feeds involved
|
|
def sort_by_feed(classifiers):
|
|
feed_classifiers = defaultdict(list)
|
|
for classifier in classifiers:
|
|
feed_classifiers[classifier.feed_id].append(classifier)
|
|
return feed_classifiers
|
|
classifier_feeds = sort_by_feed(MClassifierFeed.objects(user_id=user.pk, feed_id__in=found_feed_ids))
|
|
classifier_authors = sort_by_feed(MClassifierAuthor.objects(user_id=user.pk, feed_id__in=found_feed_ids))
|
|
classifier_titles = sort_by_feed(MClassifierTitle.objects(user_id=user.pk, feed_id__in=found_feed_ids))
|
|
classifier_tags = sort_by_feed(MClassifierTag.objects(user_id=user.pk, feed_id__in=found_feed_ids))
|
|
|
|
# Just need to format stories
|
|
for story in stories:
|
|
story_date = localtime_for_timezone(story['story_date'], user.profile.timezone)
|
|
now = localtime_for_timezone(datetime.datetime.now(), user.profile.timezone)
|
|
story['short_parsed_date'] = format_story_link_date__short(story_date, now)
|
|
story['long_parsed_date'] = format_story_link_date__long(story_date, now)
|
|
story['read_status'] = 0
|
|
if story['id'] in starred_stories:
|
|
story['starred'] = True
|
|
starred_date = localtime_for_timezone(starred_stories[story['id']], user.profile.timezone)
|
|
story['starred_date'] = format_story_link_date__long(starred_date, now)
|
|
story['intelligence'] = {
|
|
'feed': apply_classifier_feeds(classifier_feeds[story['story_feed_id']], story['story_feed_id']),
|
|
'author': apply_classifier_authors(classifier_authors[story['story_feed_id']], story),
|
|
'tags': apply_classifier_tags(classifier_tags[story['story_feed_id']], story),
|
|
'title': apply_classifier_titles(classifier_titles[story['story_feed_id']], story),
|
|
}
|
|
|
|
diff = datetime.datetime.utcnow() - start
|
|
timediff = float("%s.%.2s" % (diff.seconds, (diff.microseconds / 1000)))
|
|
logging.info(" ---> [%s] ~FCLoading river stories: page %s - ~SB%s/%s stories ~SN(%s/%s/%s feeds) ~FB(%s seconds)" % (
|
|
request.user, page, len(stories), len(mstories), len(found_feed_ids), len(feed_ids), len(original_feed_ids), timediff))
|
|
|
|
return dict(stories=stories)
|
|
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def mark_all_as_read(request):
|
|
code = 1
|
|
days = int(request.POST['days'])
|
|
|
|
feeds = UserSubscription.objects.filter(user=request.user)
|
|
for sub in feeds:
|
|
if days == 0:
|
|
sub.mark_feed_read()
|
|
else:
|
|
read_date = datetime.datetime.utcnow() - datetime.timedelta(days=days)
|
|
if sub.mark_read_date < read_date:
|
|
sub.needs_unread_recalc = True
|
|
sub.mark_read_date = read_date
|
|
sub.save()
|
|
|
|
logging.info(" ---> [%s] ~FMMarking all as read: ~SB%s days" % (request.user, days,))
|
|
return dict(code=code)
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def mark_story_as_read(request):
|
|
story_ids = request.REQUEST.getlist('story_id')
|
|
feed_id = int(request.REQUEST['feed_id'])
|
|
|
|
try:
|
|
usersub = UserSubscription.objects.select_related('feed').get(user=request.user, feed=feed_id)
|
|
except Feed.DoesNotExist:
|
|
duplicate_feed = DuplicateFeed.objects.filter(duplicate_feed_id=feed_id)
|
|
if duplicate_feed:
|
|
try:
|
|
usersub = UserSubscription.objects.get(user=request.user,
|
|
feed=duplicate_feed[0].feed)
|
|
except Feed.DoesNotExist:
|
|
return dict(code=-1)
|
|
|
|
if not usersub.needs_unread_recalc:
|
|
usersub.needs_unread_recalc = True
|
|
usersub.save()
|
|
|
|
data = dict(code=0, payload=story_ids)
|
|
|
|
if len(story_ids) > 1:
|
|
logging.info(" ---> [%s] ~FYRead %s stories in feed: %s" % (request.user, len(story_ids), usersub.feed))
|
|
else:
|
|
logging.info(" ---> [%s] ~FYRead story in feed: %s" % (request.user, usersub.feed))
|
|
|
|
for story_id in story_ids:
|
|
try:
|
|
story = MStory.objects(story_feed_id=feed_id, story_guid=story_id)[0]
|
|
except IndexError:
|
|
# Story has been deleted, probably by feed_fetcher.
|
|
continue
|
|
now = datetime.datetime.utcnow()
|
|
m = MUserStory(story=story, user_id=request.user.pk, feed_id=feed_id, read_date=now)
|
|
try:
|
|
m.save()
|
|
except OperationError:
|
|
logging.info(' ---> [%s] ~BRMarked story as read: Duplicate Story -> %s' % (request.user, story_id))
|
|
|
|
return data
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def mark_story_as_unread(request):
|
|
story_id = request.POST['story_id']
|
|
feed_id = int(request.POST['feed_id'])
|
|
|
|
try:
|
|
usersub = UserSubscription.objects.select_related('feed').get(user=request.user, feed=feed_id)
|
|
except Feed.DoesNotExist:
|
|
duplicate_feed = DuplicateFeed.objects.filter(duplicate_feed_id=feed_id)
|
|
if duplicate_feed:
|
|
try:
|
|
usersub = UserSubscription.objects.get(user=request.user,
|
|
feed=duplicate_feed[0].feed)
|
|
except Feed.DoesNotExist:
|
|
return dict(code=-1)
|
|
|
|
if not usersub.needs_unread_recalc:
|
|
usersub.needs_unread_recalc = True
|
|
usersub.save()
|
|
|
|
data = dict(code=0, payload=dict(story_id=story_id))
|
|
logging.info(" ---> [%s] ~FY~SBUnread~SN story in feed: %s" % (request.user, usersub.feed))
|
|
|
|
story = MStory.objects(story_feed_id=feed_id, story_guid=story_id)[0]
|
|
m = MUserStory.objects(story=story, user_id=request.user.pk, feed_id=feed_id)
|
|
m.delete()
|
|
|
|
return data
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def mark_feed_as_read(request):
|
|
feed_ids = request.REQUEST.getlist('feed_id')
|
|
code = 0
|
|
for feed_id in feed_ids:
|
|
try:
|
|
feed = Feed.objects.get(id=feed_id)
|
|
except Feed.DoesNotExist:
|
|
continue
|
|
code = 0
|
|
|
|
us = UserSubscription.objects.get(feed=feed, user=request.user)
|
|
try:
|
|
us.mark_feed_read()
|
|
except IntegrityError:
|
|
code = -1
|
|
else:
|
|
code = 1
|
|
|
|
logging.info(" ---> [%s] ~FMMarking feed as read: ~SB%s" % (request.user, feed,))
|
|
MUserStory.objects(user_id=request.user.pk, feed_id=feed_id).delete()
|
|
return dict(code=code)
|
|
|
|
def _parse_user_info(user):
|
|
return {
|
|
'user_info': {
|
|
'is_anonymous': json.encode(user.is_anonymous()),
|
|
'is_authenticated': json.encode(user.is_authenticated()),
|
|
'username': json.encode(user.username if user.is_authenticated() else 'Anonymous')
|
|
}
|
|
}
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def add_url(request):
|
|
code = 0
|
|
url = request.POST['url']
|
|
folder = request.POST['folder']
|
|
|
|
code, message, _ = UserSubscription.add_subscription(user=request.user, feed_address=url, folder=folder)
|
|
|
|
return dict(code=code, message=message)
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def add_folder(request):
|
|
folder = request.POST['folder']
|
|
parent_folder = request.POST['parent_folder']
|
|
|
|
logging.info(" ---> [%s] ~FRAdding Folder: ~SB%s (in %s)" % (request.user, folder, parent_folder))
|
|
|
|
if folder:
|
|
code = 1
|
|
message = ""
|
|
user_sub_folders_object, _ = UserSubscriptionFolders.objects.get_or_create(user=request.user)
|
|
if user_sub_folders_object.folders:
|
|
user_sub_folders = json.decode(user_sub_folders_object.folders)
|
|
else:
|
|
user_sub_folders = []
|
|
obj = {folder: []}
|
|
user_sub_folders = add_object_to_folder(obj, parent_folder, user_sub_folders)
|
|
user_sub_folders_object.folders = json.encode(user_sub_folders)
|
|
user_sub_folders_object.save()
|
|
else:
|
|
code = -1
|
|
message = "Gotta write in a folder name."
|
|
|
|
return dict(code=code, message=message)
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def delete_feed(request):
|
|
feed_id = int(request.POST['feed_id'])
|
|
in_folder = request.POST.get('in_folder', '')
|
|
|
|
user_sub_folders = get_object_or_404(UserSubscriptionFolders, user=request.user)
|
|
user_sub_folders.delete_feed(feed_id, in_folder)
|
|
|
|
return dict(code=1)
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def delete_folder(request):
|
|
folder_to_delete = request.POST['folder_name']
|
|
in_folder = request.POST.get('in_folder', '')
|
|
feed_ids_in_folder = request.REQUEST.getlist('feed_id')
|
|
feed_ids_in_folder = [int(f) for f in feed_ids_in_folder if f]
|
|
|
|
# Works piss poor with duplicate folder titles, if they are both in the same folder.
|
|
# Deletes all, but only in the same folder parent. But nobody should be doing that, right?
|
|
user_sub_folders = get_object_or_404(UserSubscriptionFolders, user=request.user)
|
|
user_sub_folders.delete_folder(folder_to_delete, in_folder, feed_ids_in_folder)
|
|
|
|
return dict(code=1)
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def rename_feed(request):
|
|
feed = get_object_or_404(Feed, pk=int(request.POST['feed_id']))
|
|
user_sub = UserSubscription.objects.get(user=request.user, feed=feed)
|
|
feed_title = request.POST['feed_title']
|
|
|
|
logging.info(" ---> [%s] ~FRRenaming feed '~SB%s~SN' to: ~SB%s" % (
|
|
request.user, feed.feed_title, feed_title))
|
|
|
|
user_sub.user_title = feed_title
|
|
user_sub.save()
|
|
|
|
return dict(code=1)
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def rename_folder(request):
|
|
folder_to_rename = request.POST['folder_name']
|
|
new_folder_name = request.POST['new_folder_name']
|
|
in_folder = request.POST.get('in_folder', '')
|
|
|
|
# Works piss poor with duplicate folder titles, if they are both in the same folder.
|
|
# renames all, but only in the same folder parent. But nobody should be doing that, right?
|
|
if new_folder_name:
|
|
user_sub_folders = get_object_or_404(UserSubscriptionFolders, user=request.user)
|
|
user_sub_folders.rename_folder(folder_to_rename, new_folder_name, in_folder)
|
|
|
|
return dict(code=1)
|
|
|
|
@login_required
|
|
def add_feature(request):
|
|
if not request.user.is_staff:
|
|
return HttpResponseForbidden()
|
|
|
|
code = -1
|
|
form = FeatureForm(request.POST)
|
|
|
|
if form.is_valid():
|
|
form.save()
|
|
code = 1
|
|
return HttpResponseRedirect(reverse('index'))
|
|
|
|
return dict(code=code)
|
|
|
|
@json.json_view
|
|
def load_features(request):
|
|
page = int(request.POST.get('page', 0))
|
|
logging.info(" ---> [%s] ~FBBrowse features: Page #%s" % (request.user, page+1))
|
|
features = Feature.objects.all()[page*3:(page+1)*3+1].values()
|
|
features = [{
|
|
'description': f['description'],
|
|
'date': f['date'].strftime("%b %d, %Y")
|
|
} for f in features]
|
|
return features
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def save_feed_order(request):
|
|
folders = request.POST.get('folders')
|
|
if folders:
|
|
# Test that folders can be JSON decoded
|
|
folders_list = json.decode(folders)
|
|
assert folders_list is not None
|
|
logging.info(" ---> [%s] ~FBFeed re-ordering: ~SB%s folders/feeds" % (request.user,
|
|
len(folders_list)))
|
|
user_sub_folders = UserSubscriptionFolders.objects.get(user=request.user)
|
|
user_sub_folders.folders = folders
|
|
user_sub_folders.save()
|
|
|
|
return {}
|
|
|
|
@json.json_view
|
|
def get_feeds_trainer(request):
|
|
classifiers = []
|
|
feed_id = request.POST.get('feed_id')
|
|
user = get_user(request)
|
|
usersubs = UserSubscription.objects.filter(user=user, active=True)
|
|
if feed_id:
|
|
feed = get_object_or_404(Feed, pk=feed_id)
|
|
usersubs = usersubs.filter(feed=feed)
|
|
usersubs = usersubs.select_related('feed').order_by('-feed__stories_last_month')
|
|
|
|
for us in usersubs:
|
|
if (not us.is_trained and us.feed.stories_last_month > 0) or feed_id:
|
|
classifier = dict()
|
|
classifier['classifiers'] = get_classifiers_for_user(user, us.feed.pk)
|
|
classifier['feed_id'] = us.feed.pk
|
|
classifier['stories_last_month'] = us.feed.stories_last_month
|
|
classifier['feed_tags'] = json.decode(us.feed.data.popular_tags) if us.feed.data.popular_tags else []
|
|
classifier['feed_authors'] = json.decode(us.feed.data.popular_authors) if us.feed.data.popular_authors else []
|
|
classifiers.append(classifier)
|
|
|
|
logging.info(" ---> [%s] ~FGLoading Trainer: ~SB%s feeds" % (user, len(classifiers)))
|
|
|
|
return classifiers
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def save_feed_chooser(request):
|
|
approved_feeds = [int(feed_id) for feed_id in request.POST.getlist('approved_feeds') if feed_id][:64]
|
|
activated = 0
|
|
usersubs = UserSubscription.objects.filter(user=request.user)
|
|
|
|
for sub in usersubs:
|
|
try:
|
|
if sub.feed.pk in approved_feeds:
|
|
sub.active = True
|
|
activated += 1
|
|
sub.save()
|
|
sub.feed.count_subscribers()
|
|
elif sub.active:
|
|
sub.active = False
|
|
sub.save()
|
|
except Feed.DoesNotExist:
|
|
pass
|
|
|
|
|
|
logging.info(' ---> [%s] ~BB~FW~SBActivated standard account: ~FC%s~SN/~SB%s' % (request.user,
|
|
activated,
|
|
usersubs.count()))
|
|
request.user.profile.queue_new_feeds()
|
|
request.user.profile.refresh_stale_feeds(exclude_new=True)
|
|
|
|
return {'activated': activated}
|
|
|
|
@ajax_login_required
|
|
def retrain_all_sites(request):
|
|
for sub in UserSubscription.objects.filter(user=request.user):
|
|
sub.is_trained = False
|
|
sub.save()
|
|
|
|
return get_feeds_trainer(request)
|
|
|
|
@login_required
|
|
def activate_premium_account(request):
|
|
try:
|
|
usersubs = UserSubscription.objects.select_related('feed').filter(user=request.user)
|
|
for sub in usersubs:
|
|
sub.active = True
|
|
sub.save()
|
|
if sub.feed.premium_subscribers <= 0:
|
|
sub.feed.count_subscribers()
|
|
sub.feed.schedule_feed_fetch_immediately()
|
|
except Exception, e:
|
|
subject = "Premium activation failed"
|
|
message = "%s -- %s\n\n%s" % (request.user, usersubs, e)
|
|
mail_admins(subject, message, fail_silently=True)
|
|
|
|
request.user.profile.is_premium = True
|
|
request.user.profile.save()
|
|
|
|
return HttpResponseRedirect(reverse('index'))
|
|
|
|
@login_required
|
|
def login_as(request):
|
|
if not request.user.is_staff:
|
|
logging.info(' ---> [%s] ~SKNON-STAFF LOGGING IN AS ANOTHER USER!' % request.user)
|
|
assert False
|
|
return HttpResponseForbidden()
|
|
username = request.GET['user']
|
|
user = get_object_or_404(User, username=username)
|
|
user.backend = settings.AUTHENTICATION_BACKENDS[0]
|
|
login_user(request, user)
|
|
return HttpResponseRedirect(reverse('index'))
|
|
|
|
def iframe_buster(request):
|
|
logging.info(" ---> [%s] ~FB~SBiFrame bust!" % (request.user,))
|
|
return HttpResponse(status=204)
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def mark_story_as_starred(request):
|
|
code = 1
|
|
feed_id = int(request.POST['feed_id'])
|
|
story_id = request.POST['story_id']
|
|
|
|
story = MStory.objects(story_feed_id=feed_id, story_guid=story_id).limit(1)
|
|
if story:
|
|
story_db = dict([(k, v) for k, v in story[0]._data.items()
|
|
if k is not None and v is not None])
|
|
now = datetime.datetime.now()
|
|
story_values = dict(user_id=request.user.pk, starred_date=now, **story_db)
|
|
MStarredStory.objects.create(**story_values)
|
|
logging.info(' ---> [%s] ~FCStarring: ~SB%s' % (request.user, story[0].story_title[:50]))
|
|
else:
|
|
code = -1
|
|
|
|
return {'code': code}
|
|
|
|
@ajax_login_required
|
|
@json.json_view
|
|
def mark_story_as_unstarred(request):
|
|
code = 1
|
|
story_id = request.POST['story_id']
|
|
|
|
starred_story = MStarredStory.objects(user_id=request.user.pk, story_guid=story_id)
|
|
if starred_story:
|
|
logging.info(' ---> [%s] ~FCUnstarring: ~SB%s' % (request.user, starred_story[0].story_title[:50]))
|
|
starred_story.delete()
|
|
else:
|
|
code = -1
|
|
|
|
return {'code': code}
|
|
|