mirror of
https://github.com/viq/NewsBlur.git
synced 2025-04-13 09:38:09 +00:00
24 lines
No EOL
1.1 KiB
Python
24 lines
No EOL
1.1 KiB
Python
from django.db import models
|
|
from apps.rss_feeds.models import DuplicateFeed
|
|
from utils import log as logging
|
|
|
|
class UserSubscriptionManager(models.Manager):
|
|
def get(self, *args, **kwargs):
|
|
try:
|
|
return super(UserSubscriptionManager, self).get(*args, **kwargs)
|
|
except:
|
|
if isinstance(kwargs.get('feed'), int):
|
|
feed_id = kwargs.get('feed')
|
|
elif 'feed' in kwargs:
|
|
feed_id = kwargs['feed'].pk
|
|
elif 'feed__pk' in kwargs:
|
|
feed_id = kwargs['feed__pk']
|
|
dupe_feed = DuplicateFeed.objects.filter(duplicate_feed_id=feed_id)
|
|
if dupe_feed:
|
|
feed = dupe_feed[0].feed
|
|
if 'feed' in kwargs:
|
|
kwargs['feed'] = feed
|
|
elif 'feed__pk' in kwargs:
|
|
kwargs['feed__pk'] = feed.pk
|
|
logging.debug(" ---> [%s] ~BRFound dupe UserSubscription: ~SB%s (%s)" % (getattr(kwargs.get('user'), 'username'), feed, feed_id))
|
|
return super(UserSubscriptionManager, self).get(*args, **kwargs) |