mirror of
https://github.com/nikdoof/test-auth.git
synced 2025-12-14 14:52:15 +00:00
97 lines
2.9 KiB
Python
97 lines
2.9 KiB
Python
from datetime import datetime, date
|
|
import urllib
|
|
|
|
from django.utils import simplejson as json
|
|
from django.db import models
|
|
from django.contrib.auth.models import User
|
|
from django.utils.timezone import now, utc
|
|
|
|
from reddit.api import Comment
|
|
|
|
|
|
class RedditAccount(models.Model):
|
|
"""
|
|
Represents a User ID on Reddit
|
|
|
|
This model can be populated by API update:
|
|
|
|
>>> from reddit.models import RedditAccount
|
|
>>> mod = RedditAccount()
|
|
>>> mod.username = 'nik_doof'
|
|
>>> mod.api_update()
|
|
>>> mod.reddit_id
|
|
u'1axok'
|
|
"""
|
|
|
|
user = models.ForeignKey(User, blank=True, null=True)
|
|
|
|
username = models.CharField("Reddit Username", max_length=32, blank=False, unique=True)
|
|
reddit_id = models.CharField("Reddit ID", max_length=32, help_text="Reddit's unique id for this user")
|
|
|
|
link_karma = models.IntegerField("Link Karma", help_text="Number of link karma")
|
|
comment_karma = models.IntegerField("Comment Karma", help_text="Number of comment karma")
|
|
validated = models.BooleanField("Validated", help_text="Indicates if the user has been validated")
|
|
|
|
date_created = models.DateTimeField("Date Created")
|
|
last_update = models.DateTimeField("Last Update from API")
|
|
|
|
def api_update(self):
|
|
try:
|
|
jsondoc = json.load(urllib.urlopen("http://reddit.com/user/%s/about.json" % self.username))
|
|
except:
|
|
raise self.DoesNotExist
|
|
|
|
if 'error' in jsondoc:
|
|
raise self.DoesNotExist
|
|
|
|
data = jsondoc['data']
|
|
|
|
self.link_karma = int(data['link_karma'])
|
|
self.comment_karma = int(data['comment_karma'])
|
|
self.reddit_id = unicode(data['id'], 'utf-8')
|
|
|
|
self.date_created = datetime.fromtimestamp(data['created_utc']).replace(tzinfo=utc)
|
|
self.last_update = now()
|
|
|
|
def recent_posts(self):
|
|
""" Returns the first page of posts visible on the user's profile page """
|
|
|
|
try:
|
|
jsondoc = json.load(urllib.urlopen("http://reddit.com/user/%s.json" % self.username))
|
|
except:
|
|
raise self.DoesNotExist
|
|
|
|
posts = []
|
|
for item in jsondoc['data']['children']:
|
|
if item['kind'] == 't1':
|
|
posts.append(Comment(item['data']))
|
|
elif item['kind'] == 't3':
|
|
posts.append(item['data'])
|
|
|
|
return posts
|
|
|
|
@property
|
|
def is_valid(self):
|
|
if not self.date_created:
|
|
return False
|
|
|
|
# Account 3 months old?
|
|
if (date.today() - self.date_created.date()).days >= 90:
|
|
return True
|
|
|
|
# Account created after 9/2/10 and before 13/2/10
|
|
if self.date_created.date() >= date(2010, 2, 9) and self.date_created.date() <= date(2010, 2, 13):
|
|
return True
|
|
|
|
return False
|
|
|
|
|
|
class Meta:
|
|
app_label = 'reddit'
|
|
ordering = ['username']
|
|
verbose_name = 'Reddit Account'
|
|
verbose_name_plural = 'Reddit Accounts'
|
|
|
|
def __unicode__(self):
|
|
return self.username
|