''' external library imports ''' import binascii import logging import hashlib import re import random import urllib import urllib2 from ckeditor.fields import RichTextField from datetime import timedelta, datetime from decimal import Decimal from notification import models as notification from postmonkey import PostMonkey, MailChimpException from tempfile import SpooledTemporaryFile ''' django imports ''' from django.conf import settings from django.contrib.auth.models import User from django.contrib.sites.models import Site from django.core.urlresolvers import reverse from django.core.files.base import ContentFile from django.db import models from django.db.models import F, Q, get_model from django.db.models.signals import post_save from django.utils.translation import ugettext_lazy as _ ''' regluit imports ''' import regluit import regluit.core.isbn import regluit.core.cc as cc from regluit.core.epub import personalize, ungluify, test_epub, ask_epub from regluit.core.pdf import ask_pdf, pdf_append from regluit.marc.models import MARCRecord as NewMARC from regluit.core.signals import ( successful_campaign, unsuccessful_campaign, wishlist_added ) from regluit.utils import crypto from regluit.utils.localdatetime import now, date_today from regluit.payment.parameters import ( TRANSACTION_STATUS_ACTIVE, TRANSACTION_STATUS_COMPLETE, TRANSACTION_STATUS_CANCELED, TRANSACTION_STATUS_ERROR, TRANSACTION_STATUS_FAILED, TRANSACTION_STATUS_INCOMPLETE ) from regluit.core.parameters import ( REWARDS, BUY2UNGLUE, THANKS, INDIVIDUAL, LIBRARY, BORROWED, TESTING, RESERVE, THANKED, ) from regluit.booxtream import BooXtream watermarker = BooXtream() from regluit.libraryauth.models import Library pm = PostMonkey(settings.MAILCHIMP_API_KEY) logger = logging.getLogger(__name__) class UnglueitError(RuntimeError): pass class Key(models.Model): """an encrypted key store""" name = models.CharField(max_length=255, unique=True) encrypted_value = models.TextField(null=True, blank=True) def _get_value(self): return crypto.decrypt_string(binascii.a2b_hex(self.encrypted_value), settings.SECRET_KEY) def _set_value(self, value): self.encrypted_value = binascii.b2a_hex(crypto.encrypt_string(value, settings.SECRET_KEY)) value = property(_get_value, _set_value) def __unicode__(self): return "Key with name {0}".format(self.name) class CeleryTask(models.Model): created = models.DateTimeField(auto_now_add=True, default=now()) task_id = models.CharField(max_length=255) user = models.ForeignKey(User, related_name="tasks", null=True) description = models.CharField(max_length=2048, null=True) # a description of what the task is function_name = models.CharField(max_length=1024) # used to reconstitute the AsyncTask with which to get status function_args = models.IntegerField(null=True) # not full generalized here -- takes only a single arg for now. active = models.NullBooleanField(default=True) def __unicode__(self): return "Task %s arg:%s ID# %s %s: State %s " % (self.function_name, self.function_args, self.task_id, self.description, self.state) @property def AsyncResult(self): f = getattr(regluit.core.tasks,self.function_name) return f.AsyncResult(self.task_id) @property def state(self): f = getattr(regluit.core.tasks,self.function_name) return f.AsyncResult(self.task_id).state @property def result(self): f = getattr(regluit.core.tasks,self.function_name) return f.AsyncResult(self.task_id).result @property def info(self): f = getattr(regluit.core.tasks,self.function_name) return f.AsyncResult(self.task_id).info class Claim(models.Model): STATUSES = (( u'active', u'Claim has been accepted.'), (u'pending', u'Claim is pending acceptance.'), (u'release', u'Claim has not been accepted.'), ) created = models.DateTimeField(auto_now_add=True) rights_holder = models.ForeignKey("RightsHolder", related_name="claim", null=False ) work = models.ForeignKey("Work", related_name="claim", null=False ) user = models.ForeignKey(User, related_name="claim", null=False ) status = models.CharField(max_length=7, choices=STATUSES, default='active') @property def can_open_new(self): # whether a campaign can be opened for this claim #must be an active claim if self.status != 'active': return False #can't already be a campaign for campaign in self.campaigns: if campaign.status in ['ACTIVE','INITIALIZED', 'SUCCESSFUL']: return False return True def __unicode__(self): return self.work.title @property def campaign(self): return self.work.last_campaign() @property def campaigns(self): return self.work.campaigns.all() def notify_claim(sender, created, instance, **kwargs): if 'example.org' in instance.user.email or hasattr(instance,'dont_notify'): return try: (rights, new_rights) = User.objects.get_or_create(email='rights@gluejar.com',defaults={'username':'RightsatUnglueit'}) except: rights = None if instance.user == instance.rights_holder.owner: ul=(instance.user, rights) else: ul=(instance.user, instance.rights_holder.owner, rights) notification.send(ul, "rights_holder_claim", {'claim': instance,}) post_save.connect(notify_claim,sender=Claim) class RightsHolder(models.Model): created = models.DateTimeField(auto_now_add=True) email = models.CharField(max_length=100, blank=True) rights_holder_name = models.CharField(max_length=100, blank=False) owner = models.ForeignKey(User, related_name="rights_holder", null=False ) can_sell = models.BooleanField(default=False) def __unicode__(self): return self.rights_holder_name class Premium(models.Model): PREMIUM_TYPES = ((u'00', u'Default'),(u'CU', u'Custom'),(u'XX', u'Inactive')) TIERS = {"supporter":25, "patron":50, "bibliophile":100} #should load this from fixture created = models.DateTimeField(auto_now_add=True) type = models.CharField(max_length=2, choices=PREMIUM_TYPES) campaign = models.ForeignKey("Campaign", related_name="premiums", null=True) amount = models.DecimalField(max_digits=10, decimal_places=0, blank=False) description = models.TextField(null=True, blank=False) limit = models.IntegerField(default = 0) @property def premium_count(self): t_model=get_model('payment','Transaction') return t_model.objects.filter(premium=self).count() @property def premium_remaining(self): t_model=get_model('payment','Transaction') return self.limit - t_model.objects.filter(premium=self).count() def __unicode__(self): return (self.campaign.work.title if self.campaign else '') + ' $' + str(self.amount) class PledgeExtra: def __init__(self,premium=None,anonymous=False,ack_name='',ack_dedication='',offer=None): self.anonymous = anonymous self.premium = premium self.offer = offer self.extra = {} if ack_name: self.extra['ack_name']=ack_name if ack_dedication: self.extra['ack_dedication']=ack_dedication class CampaignAction(models.Model): timestamp = models.DateTimeField(auto_now_add=True) # anticipated types: activated, withdrawn, suspended, restarted, succeeded, failed, unglued type = models.CharField(max_length=15) comment = models.TextField(null=True, blank=True) campaign = models.ForeignKey("Campaign", related_name="actions", null=False) class Offer(models.Model): CHOICES = ((INDIVIDUAL,'Individual license'),(LIBRARY,'Library License')) work = models.ForeignKey("Work", related_name="offers", null=False) price = models.DecimalField(max_digits=6, decimal_places=2, null=True, blank=False) license = models.PositiveSmallIntegerField(null = False, default = INDIVIDUAL, choices=CHOICES) active = models.BooleanField(default=False) @property def days_per_copy(self): return Decimal(float(self.price) / self.work.last_campaign().dollar_per_day ) @property def get_thanks_display(self): if self.license == LIBRARY: return 'Suggested contribution for libraries' else: return 'Suggested contribution for individuals' class Acq(models.Model): """ Short for Acquisition, this is a made-up word to describe the thing you acquire when you buy or borrow an ebook """ CHOICES = ((INDIVIDUAL,'Individual license'),(LIBRARY,'Library License'),(BORROWED,'Borrowed from Library'), (TESTING,'Just for Testing'), (RESERVE,'On Reserve'),(THANKED,'Already Thanked'),) created = models.DateTimeField(auto_now_add=True, db_index=True,) expires = models.DateTimeField(null=True) refreshes = models.DateTimeField(auto_now_add=True, default=now()) refreshes.editable=True refreshed = models.BooleanField(default=True) work = models.ForeignKey("Work", related_name='acqs', null=False) user = models.ForeignKey(User, related_name='acqs') license = models.PositiveSmallIntegerField(null = False, default = INDIVIDUAL, choices=CHOICES) watermarked = models.ForeignKey("booxtream.Boox", null=True) nonce = models.CharField(max_length=32, null=True) # when the acq is a loan, this points at the library's acq it's derived from lib_acq = models.ForeignKey("self", related_name="loans", null=True) def __unicode__(self): if self.lib_acq: return "%s, %s: %s for %s" % (self.work, self.get_license_display(), self.lib_acq.user, self.user) else: return "%s, %s for %s" % (self.work, self.get_license_display(), self.user,) @property def expired(self): if self.expires is None: return False else: return self.expires < datetime.now() def get_mobi_url(self): if self.expired: return '' return self.get_watermarked().download_link_mobi def get_epub_url(self): if self.expired: return '' return self.get_watermarked().download_link_epub def get_watermarked(self): if self.watermarked == None or self.watermarked.expired: if self.on_reserve: self.borrow(self.user) do_watermark= self.work.last_campaign().do_watermark params={ 'customeremailaddress': self.user.email if do_watermark else '', 'customername': self.user.username if do_watermark else 'an ungluer', 'languagecode':'1033', 'expirydays': 1, 'downloadlimit': 7, 'exlibris':0, 'chapterfooter': 0, 'disclaimer':0, 'referenceid': '%s:%s:%s' % (self.work.id, self.user.id, self.id) if do_watermark else 'N/A', 'kf8mobi': True, 'epub': True, } personalized = personalize(self.work.epubfiles()[0].file, self) personalized.seek(0) self.watermarked = watermarker.platform(epubfile= personalized, **params) self.save() return self.watermarked def _hash(self): return hashlib.md5('%s:%s:%s:%s'%(settings.TWITTER_CONSUMER_SECRET,self.user.id,self.work.id,self.created)).hexdigest() def expire_in(self, delta): self.expires = now() + delta self.save() if self.lib_acq: self.lib_acq.refreshes = now() + delta self.lib_acq.refreshed = False self.lib_acq.save() @property def on_reserve(self): return self.license==RESERVE def borrow(self, user=None): if self.on_reserve: self.license=BORROWED self.expire_in(timedelta(days=14)) self.user.wishlist.add_work( self.work, "borrow") notification.send([self.user], "library_borrow", {'acq':self}) return self elif self.borrowable and user: user.wishlist.add_work( self.work, "borrow") borrowed = Acq.objects.create(user=user,work=self.work,license= BORROWED, lib_acq=self) from regluit.core.tasks import watermark_acq notification.send([user], "library_borrow", {'acq':borrowed}) watermark_acq.delay(borrowed) return borrowed @property def borrowable(self): if self.license == RESERVE and not self.expired: return True if self.license == LIBRARY: return self.refreshes < datetime.now() else: return False @property def holds(self): return Hold.objects.filter(library__user=self.user,work=self.work).order_by('created') def config_acq(sender, instance, created, **kwargs): if created: instance.nonce=instance._hash() instance.save() if instance.license == RESERVE: instance.expire_in(timedelta(hours=24)) if instance.license == BORROWED: instance.expire_in(timedelta(days=14)) post_save.connect(config_acq,sender=Acq) class Hold(models.Model): created = models.DateTimeField(auto_now_add=True) work = models.ForeignKey("Work", related_name='holds', null=False) user = models.ForeignKey(User, related_name='holds', null=False) library = models.ForeignKey(Library, related_name='holds', null=False) def __unicode__(self): return '%s for %s at %s' % (self.work,self.user.username,self.library) def ahead(self): return Hold.objects.filter(work=self.work,library=self.library,created__lt=self.created).count() class Campaign(models.Model): LICENSE_CHOICES = cc.FREECHOICES created = models.DateTimeField(auto_now_add=True,) name = models.CharField(max_length=500, null=True, blank=False) description = RichTextField(null=True, blank=False) details = RichTextField(null=True, blank=True) target = models.DecimalField(max_digits=14, decimal_places=2, null=True, default=0.00) license = models.CharField(max_length=255, choices = LICENSE_CHOICES, default='CC BY-NC-ND') left = models.DecimalField(max_digits=14, decimal_places=2, null=True, db_index=True,) deadline = models.DateTimeField(db_index=True, null=True) dollar_per_day = models.FloatField(null=True) cc_date_initial = models.DateTimeField(null=True) activated = models.DateTimeField(null=True, db_index=True,) paypal_receiver = models.CharField(max_length=100, blank=True) amazon_receiver = models.CharField(max_length=100, blank=True) work = models.ForeignKey("Work", related_name="campaigns", null=False) managers = models.ManyToManyField(User, related_name="campaigns", null=False) # status: INITIALIZED, ACTIVE, SUSPENDED, WITHDRAWN, SUCCESSFUL, UNSUCCESSFUL status = models.CharField(max_length=15, null=True, blank=False, default="INITIALIZED", db_index=True,) type = models.PositiveSmallIntegerField(null = False, default = REWARDS, choices=((REWARDS,'Pledge-to-unglue campaign'),(BUY2UNGLUE,'Buy-to-unglue campaign'),(THANKS,'Thanks-for-ungluing campaign'))) edition = models.ForeignKey("Edition", related_name="campaigns", null=True) email = models.CharField(max_length=100, blank=True) publisher = models.ForeignKey("Publisher", related_name="campaigns", null=True) do_watermark = models.BooleanField(default=True) use_add_ask = models.BooleanField(default=True) def __init__(self, *args, **kwargs): self.problems=[] return super(Campaign, self).__init__(*args, **kwargs) def __unicode__(self): try: return u"Campaign for %s" % self.work.title except: return u"Campaign %s (no associated work)" % self.name def clone(self): """use a previous UNSUCCESSFUL campaign's data as the basis for a new campaign assume that B2U campaigns don't need cloning """ if self.clonable(): old_managers= self.managers.all() # copy custom premiums new_premiums= self.premiums.filter(type='CU') # setting pk to None will insert new copy http://stackoverflow.com/a/4736172/7782 self.pk = None self.status = 'INITIALIZED' # set deadline far in future -- presumably RH will set deadline to proper value before campaign launched self.deadline = date_today() + timedelta(days=int(settings.UNGLUEIT_LONGEST_DEADLINE)) # allow created, activated dates to be autoset by db self.created = None self.name = 'copy of %s' % self.name self.activated = None self.update_left() self.save() self.managers=old_managers # clone associated premiums for premium in new_premiums: premium.pk=None premium.created = None premium.campaign = self premium.save() return self else: return None def clonable(self): """campaign clonable if it's UNSUCCESSFUL and is the last campaign associated with a Work""" if self.status == 'UNSUCCESSFUL' and self.work.last_campaign().id==self.id: return True else: return False @property def launchable(self): may_launch=True try: if self.status != 'INITIALIZED': if self.status == 'ACTIVE': self.problems.append(_('The campaign is already launched')) else: self.problems.append(_('A campaign must initialized properly before it can be launched')) may_launch = False if not self.description: self.problems.append(_('A campaign must have a description')) may_launch = False if self.type==REWARDS: if self.deadline: if self.deadline.date()- date_today() > timedelta(days=int(settings.UNGLUEIT_LONGEST_DEADLINE)): self.problems.append(_('The chosen closing date is more than %s days from now' % settings.UNGLUEIT_LONGEST_DEADLINE)) may_launch = False else: self.problems.append(_('A pledge campaign must have a closing date')) may_launch = False if self.target: if self.target < Decimal(settings.UNGLUEIT_MINIMUM_TARGET): self.problems.append(_('A pledge campaign may not be launched with a target less than $%s' % settings.UNGLUEIT_MINIMUM_TARGET)) may_launch = False else: self.problems.append(_('A campaign must have a target')) may_launch = False if self.type==BUY2UNGLUE: if self.work.offers.filter(price__gt=0,active=True).count()==0: self.problems.append(_('You can\'t launch a buy-to-unglue campaign before setting a price for your ebooks' )) may_launch = False if EbookFile.objects.filter(edition__work=self.work).count()==0: self.problems.append(_('You can\'t launch a buy-to-unglue campaign if you don\'t have any ebook files uploaded' )) may_launch = False if ((self.cc_date_initial is None) or (self.cc_date_initial > datetime.combine(settings.MAX_CC_DATE, datetime.min.time())) or (self.cc_date_initial < now())): self.problems.append(_('You must set an initial Ungluing Date that is in the future and not after %s' % settings.MAX_CC_DATE )) may_launch = False if self.target: if self.target < Decimal(settings.UNGLUEIT_MINIMUM_TARGET): self.problems.append(_('A buy-to-unglue campaign may not be launched with a target less than $%s' % settings.UNGLUEIT_MINIMUM_TARGET)) may_launch = False else: self.problems.append(_('A buy-to-unglue campaign must have a target')) may_launch = False if self.type==THANKS: if EbookFile.objects.filter(edition__work=self.work).count()==0: self.problems.append(_('You can\'t launch a thanks-for-ungluing campaign if you don\'t have any ebook files uploaded' )) may_launch = False except Exception as e : self.problems.append('Exception checking launchability ' + str(e)) may_launch = False return may_launch def update_status(self, ignore_deadline_for_success=False, send_notice=False, process_transactions=False): """Updates the campaign's status. returns true if updated. for REWARDS: Computes UNSUCCESSFUL only after the deadline has passed Computes SUCCESSFUL only after the deadline has passed if ignore_deadline_for_success is TRUE -- otherwise looks just at amount of pledges accumulated by default, send_notice is False so that we have to explicitly send specify delivery of successful_campaign notice for BUY2UNGLUE: Sets SUCCESSFUL when cc_date is in the past. if process_transactions is True, also execute or cancel associated transactions """ if not self.status=='ACTIVE': return False elif self.type==REWARDS: if (ignore_deadline_for_success or self.deadline < now()) and self.current_total >= self.target: self.status = 'SUCCESSFUL' self.save() action = CampaignAction(campaign=self, type='succeeded', comment = self.current_total) action.save() if process_transactions: p = PaymentManager() results = p.execute_campaign(self) if send_notice: successful_campaign.send(sender=None,campaign=self) # should be more sophisticated in whether to return True -- look at all the transactions? return True elif self.deadline < now() and self.current_total < self.target: self.status = 'UNSUCCESSFUL' self.save() action = CampaignAction(campaign=self, type='failed', comment = self.current_total) action.save() if process_transactions: p = PaymentManager() results = p.cancel_campaign(self) if send_notice: regluit.core.signals.unsuccessful_campaign.send(sender=None,campaign=self) # should be more sophisticated in whether to return True -- look at all the transactions? return True elif self.type==BUY2UNGLUE: if self.cc_date < date_today(): self.status = 'SUCCESSFUL' self.save() action = CampaignAction(campaign=self, type='succeeded', comment = self.current_total) action.save() self.watermark_success() if send_notice: successful_campaign.send(sender=None,campaign=self) # should be more sophisticated in whether to return True -- look at all the transactions? return True return False _current_total = None @property def current_total(self): if self._current_total is None: p = PaymentManager() self._current_total = p.query_campaign(self,summary=True, campaign_total=True) return self._current_total def set_dollar_per_day(self): if self.status!='INITIALIZED' and self.dollar_per_day: return self.dollar_per_day if self.cc_date_initial is None: return None start_datetime= self.activated if self.activated else datetime.today() time_to_cc = self.cc_date_initial - start_datetime self.dollar_per_day = float(self.target)/float(time_to_cc.days) if self.status!='DEMO': self.save() return self.dollar_per_day def set_cc_date_initial(self, a_date=settings.MAX_CC_DATE): self.cc_date_initial = datetime.combine(a_date, datetime.min.time()) @property def cc_date(self): if self.type in { REWARDS, THANKS }: return None if self.dollar_per_day == None: return self.cc_date_initial.date() cc_advance_days = float(self.current_total) / self.dollar_per_day return (self.cc_date_initial-timedelta(days=cc_advance_days)).date() def update_left(self): self._current_total=None if self.type == THANKS: self.left == Decimal(0.00) elif self.type == BUY2UNGLUE: self.left = Decimal(self.dollar_per_day*float((self.cc_date_initial - datetime.today()).days))-self.current_total else: self.left = self.target - self.current_total if self.status != 'DEMO': self.save() def transactions(self, **kwargs): p = PaymentManager() # handle default parameter values kw = {'summary':False, 'campaign_total':True} kw.update(kwargs) return p.query_campaign(self, **kw) def activate(self): status = self.status if status != 'INITIALIZED': raise UnglueitError(_('Campaign needs to be initialized in order to be activated')) try: active_claim = self.work.claim.filter(status="active")[0] except IndexError, e: raise UnglueitError(_('Campaign needs to have an active claim in order to be activated')) if not self.launchable: raise UnglueitError('Configuration issues need to be addressed before campaign is activated: %s' % unicode(self.problems[0])) self.status= 'ACTIVE' self.left = self.target self.activated = datetime.today() if self.type == THANKS: # make ebooks from ebookfiles self.work.make_ebooks_from_ebfs() self.save() action = CampaignAction( campaign = self, type='activated', comment = self.get_type_display()) ungluers = self.work.wished_by() notification.queue(ungluers, "wishlist_active", {'campaign':self}, True) return self def suspend(self, reason): status = self.status if status != 'ACTIVE': raise UnglueitError(_('Campaign needs to be active in order to be suspended')) action = CampaignAction( campaign = self, type='suspended', comment = reason) action.save() self.status='SUSPENDED' self.save() return self def withdraw(self, reason): status = self.status if status != 'ACTIVE': raise UnglueitError(_('Campaign needs to be active in order to be withdrawn')) action = CampaignAction( campaign = self, type='withdrawn', comment = reason) action.save() self.status='WITHDRAWN' self.save() return self def resume(self, reason): """Change campaign status from SUSPENDED to ACTIVE. We may want to track reason for resuming and track history""" status = self.status if status != 'SUSPENDED': raise UnglueitError(_('Campaign needs to be suspended in order to be resumed')) if not reason: reason='' action = CampaignAction( campaign = self, type='restarted', comment = reason) action.save() self.status= 'ACTIVE' self.save() return self def supporters(self): # expensive query used in loop; stash it if hasattr(self, '_translist_'): return self._translist_ """nb: returns (distinct) supporter IDs, not supporter objects""" self._translist_ = self.transactions().filter(status=TRANSACTION_STATUS_ACTIVE).values_list('user', flat=True).distinct() return self._translist_ @property def supporters_count(self): # avoid transmitting the whole list if you don't need to; let the db do the count. active = self.transactions().filter(status=TRANSACTION_STATUS_ACTIVE).values_list('user', flat=True).distinct().count() complete = self.transactions().filter(status=TRANSACTION_STATUS_COMPLETE).values_list('user', flat=True).distinct().count() return active+complete @property def anon_count(self): # avoid transmitting the whole list if you don't need to; let the db do the count. complete = self.transactions().filter(status=TRANSACTION_STATUS_COMPLETE,user=None).count() return complete def transaction_to_recharge(self, user): """given a user, return the transaction to be recharged if there is one -- None otherwise""" # only if a campaign is SUCCESSFUL, we allow for recharged if self.status == 'SUCCESSFUL': if self.transaction_set.filter(Q(user=user) & (Q(status=TRANSACTION_STATUS_COMPLETE) | Q(status=TRANSACTION_STATUS_ACTIVE))).count(): # presence of an active or complete transaction means no transaction to recharge return None else: transactions = self.transaction_set.filter(Q(user=user) & (Q(status=TRANSACTION_STATUS_ERROR) | Q(status=TRANSACTION_STATUS_FAILED))) # assumption --that the first failed/errored transaction has the amount we need to recharge if transactions.count(): return transactions[0] else: return None else: return None def ungluers(self): # expensive query used in loop; stash it if hasattr(self, '_ungluers_'): return self._ungluers_ p = PaymentManager() ungluers={"all":[],"supporters":[], "patrons":[], "bibliophiles":[]} if self.status == "ACTIVE": translist = p.query_campaign(self, summary=False, pledged=True, authorized=True) elif self.status == "SUCCESSFUL": translist = p.query_campaign(self, summary=False, pledged=True, completed=True) else: translist = [] for transaction in translist: ungluers['all'].append(transaction.user) if not transaction.anonymous: if transaction.amount >= Premium.TIERS["bibliophile"]: ungluers['bibliophiles'].append(transaction.user) elif transaction.amount >= Premium.TIERS["patron"]: ungluers['patrons'].append(transaction.user) elif transaction.amount >= Premium.TIERS["supporter"]: ungluers['supporters'].append(transaction.user) self._ungluers_= ungluers return ungluers def ungluer_transactions(self): """ returns a list of authorized transactions for campaigns in progress, or completed transactions for successful campaigns used to build the acks page -- because ack_name, _link, _dedication adhere to transactions, it's easier to return transactions than ungluers """ p = PaymentManager() ungluers={"all":[],"supporters":[], "anon_supporters": 0, "patrons":[], "anon_patrons": 0, "bibliophiles":[]} if self.status == "ACTIVE": translist = p.query_campaign(self, summary=False, pledged=True, authorized=True) elif self.status == "SUCCESSFUL": translist = p.query_campaign(self, summary=False, pledged=True, completed=True) else: translist = [] for transaction in translist: ungluers['all'].append(transaction) if transaction.amount >= Premium.TIERS["bibliophile"]: ungluers['bibliophiles'].append(transaction) elif transaction.amount >= Premium.TIERS["patron"]: if transaction.anonymous: ungluers['anon_patrons'] += 1 else: ungluers['patrons'].append(transaction) elif transaction.amount >= Premium.TIERS["supporter"]: if transaction.anonymous: ungluers['anon_supporters'] += 1 else: ungluers['supporters'].append(transaction) return ungluers def effective_premiums(self): """returns the available premiums for the Campaign including any default premiums""" if self.type is BUY2UNGLUE: return Premium.objects.none() q = Q(campaign=self) | Q(campaign__isnull=True) return Premium.objects.filter(q).exclude(type='XX').order_by('amount') def custom_premiums(self): """returns only the active custom premiums for the Campaign""" if self.type is BUY2UNGLUE: return Premium.objects.none() return Premium.objects.filter(campaign=self).filter(type='CU').order_by('amount') @property def library_offer(self): return self._offer(LIBRARY) @property def individual_offer(self): return self._offer(INDIVIDUAL) def _offer(self, license): if self.type is REWARDS: return None try: return Offer.objects.get(work=self.work, active=True, license=license) except Offer.DoesNotExist: return None @property def ask_money(self): # true if there's an offer asking for money if self.type is REWARDS: return True try: Offer.objects.get(work=self.work, active=True, price__gt=0.00) return True except Offer.DoesNotExist: return False except Offer.MultipleObjectsReturned: return True @property def days_per_copy(self): if self.individual_offer: return Decimal(float(self.individual_offer.price) / self.dollar_per_day ) else: return Decimal(0) @property def rh(self): """returns the rights holder for an active or initialized campaign""" try: q = Q(status='ACTIVE') | Q(status='INITIALIZED') rh = self.work.claim.filter(q)[0].rights_holder return rh except: return None @property def rightsholder(self): """returns the name of the rights holder for an active or initialized campaign""" try: return self.rh.rights_holder_name except: return '' @property def license_url(self): return cc.CCLicense.url(self.license) @property def license_badge(self): return cc.CCLicense.badge(self.license) @property def success_date(self): if self.status == 'SUCCESSFUL': try: return self.actions.filter(type='succeeded')[0].timestamp except: return '' return '' def percent_of_goal(self): if self.type == THANKS: return 100 percent = 0 if(self.status == 'SUCCESSFUL' or self.status == 'ACTIVE'): if self.type == BUY2UNGLUE: percent = int(100 - 100*self.left/self.target) else: percent = int(self.current_total/self.target*100) return percent @property def countdown(self): from math import ceil if not self.deadline: return '' time_remaining = self.deadline - now() countdown = "" if time_remaining.days: countdown = "%s days" % str(time_remaining.days + 1) elif time_remaining.seconds > 3600: countdown = "%s hours" % str(time_remaining.seconds/3600 + 1) elif time_remaining.seconds > 60: countdown = "%s minutes" % str(time_remaining.seconds/60 + 1) else: countdown = "Seconds" return countdown @property def deadline_or_now(self): return self.deadline if self.deadline else now() @classmethod def latest_ending(cls): return (timedelta(days=int(settings.UNGLUEIT_LONGEST_DEADLINE)) + now()) def add_ask_to_ebfs(self, position=0): if not self.use_add_ask or self.type != THANKS : return done_formats= [] for ebf in self.work.ebookfiles().filter(asking = False).order_by('-created'): if ebf.format=='pdf' and 'pdf' not in done_formats: try: added = ask_pdf({'campaign':self, 'work':self.work, 'site':Site.objects.get_current()}) new_file = SpooledTemporaryFile() old_file = SpooledTemporaryFile() ebf.file.open() old_file.write(ebf.file.read()) if position==0: pdf_append(added, old_file, new_file) else: pdf_append(old_file, added, new_file) new_file.seek(0) new_ebf = EbookFile.objects.create(edition=ebf.edition, format='pdf', asking=True) new_ebf.file.save(path_for_file(ebf,None),ContentFile(new_file.read())) new_ebf.save() for old_ebf in self.work.ebookfiles().filter(asking = True, format='pdf').exclude(pk=new_ebf.pk): obsolete = Ebook.objects.filter(url=old_ebf.file.url) for eb in obsolete: eb.deactivate() old_ebf.delete() done_formats.append('pdf') except Exception as e: logger.error("error appending pdf ask %s" % (e)) elif ebf.format=='epub' and 'epub' not in done_formats: try: new_file = SpooledTemporaryFile() old_file = SpooledTemporaryFile() ebf.file.open() old_file.write(ebf.file.read()) new_file= ask_epub(old_file, {'campaign':self, 'work':self.work, 'site':Site.objects.get_current()}) new_file.seek(0) new_ebf = EbookFile.objects.create(edition=ebf.edition, format='epub', asking=True) new_ebf.file.save(path_for_file(ebf,None),ContentFile(new_file.read())) new_ebf.save() for old_ebf in self.work.ebookfiles().filter(asking = True, format='epub').exclude(pk=new_ebf.pk): obsolete = Ebook.objects.filter(url=old_ebf.file.url) for eb in obsolete: eb.deactivate() old_ebf.delete() done_formats.append('epub') except Exception as e: logger.error("error making epub ask %s" % (e)) self.work.make_ebooks_from_ebfs(add_ask=True) def make_unglued_ebf(self, format, watermarked): ebf=EbookFile.objects.create(edition=self.work.preferred_edition, format=format) r=urllib2.urlopen(watermarked.download_link(format)) ebf.file.save(path_for_file(ebf,None),ContentFile(r.read())) ebf.file.close() ebf.save() ebook=Ebook.objects.create( edition=self.work.preferred_edition, format=format, rights=self.license, provider="Unglue.it", url= settings.BASE_URL_SECURE + reverse('download_campaign',args=[self.work.id,format]), ) old_ebooks = Ebook.objects.exclude(pk=ebook.pk).filter( format=format, rights=self.license, provider="Unglue.it", ) for old_ebook in old_ebooks: old_ebook.deactivate() return ebook.pk def watermark_success(self): if self.status == 'SUCCESSFUL' and self.type == BUY2UNGLUE: params={ 'customeremailaddress': self.license, 'customername': 'The Public', 'languagecode':'1033', 'expirydays': 1, 'downloadlimit': 7, 'exlibris':0, 'chapterfooter':0, 'disclaimer':0, 'referenceid': '%s:%s:%s' % (self.work.id, self.id, self.license), 'kf8mobi': True, 'epub': True, } ungluified = ungluify(self.work.epubfiles()[0].file, self) ungluified.filename.seek(0) watermarked = watermarker.platform(epubfile= ungluified.filename, **params) self.make_unglued_ebf('epub', watermarked) self.make_unglued_ebf('mobi', watermarked) return True return False def is_pledge(self): return self.type==REWARDS @property def user_to_pay(self): return self.rh.owner ### for compatibility with MARC output def marc_records(self): return self.work.marc_records() class Identifier(models.Model): # olib, ltwk, goog, gdrd, thng, isbn, oclc, olwk, olib, gute, glue type = models.CharField(max_length=4, null=False) value = models.CharField(max_length=250, null=False) work = models.ForeignKey("Work", related_name="identifiers", null=False) edition = models.ForeignKey("Edition", related_name="identifiers", null=True) class Meta: unique_together = ("type", "value") @staticmethod def set(type=None, value=None, edition=None, work=None): # if there's already an id of this type for this work and edition, change it # if not, create it. if the id exists and points to something else, change it. identifier= Identifier.get_or_add(type=type, value=value, edition = edition, work=work) if identifier.work.id != work.id: identifier.work=work identifier.save() if identifier.edition and edition: if identifier.edition.id != edition.id: identifier.edition = edition identifier.save() others= Identifier.objects.filter(type=type, work=work, edition=edition).exclude(value=value) if others.count()>0: for other in others: other.delete() return identifier @staticmethod def get_or_add( type='goog', value=None, edition=None, work=None): try: return Identifier.objects.get(type=type, value=value) except Identifier.DoesNotExist: i=Identifier(type=type, value=value, edition=edition, work=work) i.save() return i def __unicode__(self): return u'{0}:{1}'.format(self.type, self.value) class Work(models.Model): created = models.DateTimeField(auto_now_add=True, db_index=True,) title = models.CharField(max_length=1000) language = models.CharField(max_length=5, default="en", null=False, db_index=True,) openlibrary_lookup = models.DateTimeField(null=True) num_wishes = models.IntegerField(default=0, db_index=True) description = models.TextField(default='', null=True, blank=True) selected_edition = models.ForeignKey("Edition", related_name = 'selected_works', null = True) earliest_publication = models.CharField(max_length=50, null=True) featured = models.DateTimeField(null=True, blank=True, db_index=True,) class Meta: ordering = ['title'] def __init__(self, *args, **kwargs): self._last_campaign = None super(Work, self).__init__(*args, **kwargs) @property def googlebooks_id(self): preferred_id=self.preferred_edition.googlebooks_id # note that there's always a preferred edition if preferred_id: return preferred_id try: return self.identifiers.filter(type='goog')[0].value except IndexError: return '' @property def googlebooks_url(self): if self.googlebooks_id: return "http://books.google.com/books?id=%s" % self.googlebooks_id else: return '' @property def goodreads_id(self): preferred_id=self.preferred_edition.goodreads_id if preferred_id: return preferred_id try: return self.identifiers.filter(type='gdrd')[0].value except IndexError: return '' @property def goodreads_url(self): return "http://www.goodreads.com/book/show/%s" % self.goodreads_id @property def librarything_id(self): try: return self.identifiers.filter(type='ltwk')[0].value except IndexError: return '' @property def librarything_url(self): return "http://www.librarything.com/work/%s" % self.librarything_id @property def openlibrary_id(self): try: return self.identifiers.filter(type='olwk')[0].value except IndexError: return '' @property def openlibrary_url(self): return "http://openlibrary.org" + self.openlibrary_id def cover_filetype(self): if self.uses_google_cover(): return 'jpeg' else: url = self.cover_image_small().lower() if url.endswith('.png'): return 'png' elif url.endswith('.gif'): return 'gif' elif url.endswith('.jpg') or url.endswith('.jpeg'): return 'jpeg' else: return 'image' def uses_google_cover(self): if self.preferred_edition and self.preferred_edition.cover_image: return False else: return self.googlebooks_id def cover_image_small(self): if self.preferred_edition and self.preferred_edition.cover_image_small(): return self.preferred_edition.cover_image_small() return "/static/images/generic_cover_larger.png" def cover_image_thumbnail(self): try: if self.preferred_edition and self.preferred_edition.cover_image_thumbnail(): return self.preferred_edition.cover_image_thumbnail() except IndexError: pass return "/static/images/generic_cover_larger.png" def authors(self): # assumes that they come out in the same order they go in! if self.preferred_edition and self.preferred_edition.authors.all().count()>0: return self.preferred_edition.authors.all() for edition in self.editions.all(): if edition.authors.all().count()>0: return edition.authors.all() return Author.objects.none() def author(self): # assumes that they come out in the same order they go in! if self.authors().count()>0: return self.authors()[0].name return '' def authors_short(self): # assumes that they come out in the same order they go in! if self.authors().count()==1: return self.authors()[0].name elif self.authors().count()==2: return "%s and %s" % (self.authors()[0].name, self.authors()[1].name) elif self.authors().count()>2: return "%s et al." % self.authors()[0].name return '' def last_campaign(self): # stash away the last campaign to prevent repeated lookups if hasattr(self, '_last_campaign_'): return self._last_campaign_ try: self._last_campaign_ = self.campaigns.order_by('-created')[0] except IndexError: self._last_campaign_ = None return self._last_campaign_ @property def preferred_edition(self): if self.selected_edition: return self.selected_edition if self.last_campaign(): if self.last_campaign().edition: self.selected_edition = self.last_campaign().edition self.save() return self.last_campaign().edition try: self.selected_edition = self.editions.all()[0] self.save() return self.selected_edition except IndexError: #should only happen if there are no editions for the work return None def last_campaign_status(self): campaign = self.last_campaign() if campaign: status = campaign.status else: if self.first_ebook(): status = "Available" else: status = "No campaign yet" return status def percent_unglued(self): status = 0 campaign = self.last_campaign() if campaign is not None: if(campaign.status == 'SUCCESSFUL'): status = 6 elif(campaign.status == 'ACTIVE'): target = float(campaign.target) if target <= 0: status = 6 else: if campaign.type == BUY2UNGLUE: status = int( 6 - 6*campaign.left/campaign.target) else: status = int(float(campaign.current_total)*6/target) if status >= 6: status = 6 return status def percent_of_goal(self): campaign = self.last_campaign() return 0 if campaign is None else campaign.percent_of_goal() def ebooks(self, all=False): if all: return Ebook.objects.filter(edition__work=self).order_by('-created') else: return Ebook.objects.filter(edition__work=self,active=True).order_by('-created') def ebookfiles(self): return EbookFile.objects.filter(edition__work=self).exclude(file='').order_by('-created') def epubfiles(self): # filter out non-epub because that's what booxtream accepts return EbookFile.objects.filter(edition__work=self, format='epub').exclude(file='').order_by('-created') def mobifiles(self): return EbookFile.objects.filter(edition__work=self, format='mobi').exclude(file='').order_by('-created') def pdffiles(self): return EbookFile.objects.filter(edition__work=self, format='pdf').exclude(file='').order_by('-created') def make_ebooks_from_ebfs(self, add_ask=True): if self.last_campaign().type != THANKS: # just to make sure that ebf's can be unglued by mistake return ebfs=EbookFile.objects.filter(edition__work=self).exclude(file='').order_by('-created') done_formats= [] for ebf in ebfs: previous_ebooks=Ebook.objects.filter(url= ebf.file.url,) try: previous_ebook = previous_ebooks[0] for eb in previous_ebooks[1:]: #housekeeping eb.deactivate() except IndexError: previous_ebook = None if ebf.format not in done_formats: if ebf.asking==add_ask or ebf.format=='mobi': if previous_ebook: previous_ebook.activate() else: ebook=Ebook.objects.get_or_create( edition=ebf.edition, format=ebf.format, rights=self.last_campaign().license, provider="Unglue.it", url= ebf.file.url, ) done_formats.append(ebf.format) elif previous_ebook: previous_ebook.deactivate() elif previous_ebook: previous_ebook.deactivate() return def remove_old_ebooks(self): old=Ebook.objects.filter(edition__work=self, active=True).order_by('-created') done_formats= [] for eb in old: if eb.format in done_formats: eb.deactivate() else: done_formats.append(eb.format) null_files=EbookFile.objects.filter(edition__work=self, file='') for ebf in null_files: ebf.delete() @property def download_count(self): dlc=0 for ebook in self.ebooks(all=True): dlc += ebook.download_count return dlc def first_pdf(self): return self.first_ebook('pdf') def first_epub(self): return self.first_ebook('epub') def first_pdf_url(self): try: url = self.first_ebook('pdf').url return url except: return None def first_epub_url(self): try: url = self.first_ebook('epub').url return url except: return None def first_ebook(self, ebook_format=None): if ebook_format: for ebook in self.ebooks().filter(format=ebook_format): return ebook else: for ebook in self.ebooks(): return ebook def wished_by(self): return User.objects.filter(wishlist__works__in=[self]) def update_num_wishes(self): self.num_wishes = Wishes.objects.filter(work=self).count() self.save() def first_oclc(self): if self.preferred_edition == None: return '' preferred_id=self.preferred_edition.oclc if preferred_id: return preferred_id try: return self.identifiers.filter(type='oclc')[0].value except IndexError: return '' def first_isbn_13(self): if self.preferred_edition == None: return '' preferred_id=self.preferred_edition.isbn_13 if preferred_id: return preferred_id try: return self.identifiers.filter(type='isbn')[0].value except IndexError: return '' @property def publication_date(self): if self.earliest_publication: return self.earliest_publication for edition in Edition.objects.filter(work=self, publication_date__isnull=False).order_by('publication_date'): if edition.publication_date: self.earliest_publication = edition.publication_date self.save() return edition.publication_date return '' @property def publication_date_year(self): try: return self.publication_date[:4] except IndexError: return 'unknown' def __unicode__(self): return self.title @property def has_unglued_edition(self): """ allows us to distinguish successful campaigns with ebooks still in progress from successful campaigns with ebooks available """ if self.ebooks().filter(edition__unglued=True): return True return False @property def user_with_rights(self): """ return queryset of users (should be at most one) who act for rights holders with active claims to the work """ claims = self.claim.filter(status='active') assert claims.count() < 2, "There is more than one active claim on %r" % self.title try: return claims[0].user except: return False def get_absolute_url(self): return reverse('work', args=[str(self.id)]) def publishers(self): # returns a set of publishers associated with this Work return Publisher.objects.filter(name__editions__work=self).distinct() def create_offers(self): for choice in Offer.CHOICES: if not self.offers.filter(license=choice[0]): self.offers.create(license=choice[0],active=True,price=Decimal(10)) return self.offers.all() def get_lib_license(self,user): lib_user=(lib.user for lib in user.profile.libraries) return self.get_user_license(lib_user) def borrowable(self, user): if user.is_anonymous(): return False lib_license=self.get_lib_license(user) if lib_license and lib_license.borrowable: return True return False def lib_thanked(self, user): if user.is_anonymous(): return False lib_license=self.get_lib_license(user) if lib_license and lib_license.thanked: return True return False def in_library(self,user): if user.is_anonymous(): return False lib_license=self.get_lib_license(user) if lib_license and lib_license.acqs.count(): return True return False @property def lib_acqs(self): return self.acqs.filter(license=LIBRARY) @property def test_acqs(self): return self.acqs.filter(license=TESTING).order_by('-created') class user_license: acqs=Acq.objects.none() def __init__(self,acqs): self.acqs=acqs @property def is_active(self): return self.acqs.filter(expires__isnull = True).count()>0 or self.acqs.filter(expires__gt= now()).count()>0 @property def borrowed(self): loans = self.acqs.filter(license=BORROWED,expires__gt= now()) if loans.count()==0: return None else: return loans[0] @property def purchased(self): purchases = self.acqs.filter(license=INDIVIDUAL) if purchases.count()==0: return None else: return purchases[0] @property def thanked(self): purchases = self.acqs.filter(license=THANKED) if purchases.count()==0: return None else: return purchases[0] @property def lib_acqs(self): return self.acqs.filter(license=LIBRARY) @property def next_acq(self): """ This is the next available copy in the user's libraries""" loans = self.acqs.filter(license=LIBRARY, refreshes__gt=now()).order_by('refreshes') if loans.count()==0: return None else: return loans[0] @property def borrowable(self): return self.acqs.filter(license=LIBRARY, refreshes__lt=now()).count()>0 @property def thanked(self): return self.acqs.filter(license=THANKED).count()>0 @property def borrowable_acq(self): for acq in self.acqs.filter(license=LIBRARY, refreshes__lt=now()): return acq else: return None def get_user_license(self, user): """ This is all the acqs, wrapped in user_license object for the work, user(s) """ if user==None: return None if hasattr(user, 'is_anonymous'): if user.is_anonymous(): return None return self.user_license(self.acqs.filter(user=user)) else: # assume it's several users return self.user_license(self.acqs.filter(user__in=user)) @property def has_marc(self): for record in NewMARC.objects.filter(edition__work=self): return True return False ### for compatibility with MARC output def marc_records(self): record_list = [] record_list.extend(NewMARC.objects.filter(edition__work=self)) for obj in record_list: break else: for ebook in self.ebooks(): record_list.append(ebook.edition) break return record_list class Author(models.Model): created = models.DateTimeField(auto_now_add=True) name = models.CharField(max_length=500) editions = models.ManyToManyField("Edition", related_name="authors") def __unicode__(self): return self.name @property def last_name_first(self): names = self.name.rsplit() if len(names) == 0: return '' elif len(names) == 1: return names[0] elif len(names) == 2: return names[1] + ", " + names[0] else: reversed_name= names[-1]+"," for name in names[0:-1]: reversed_name+=" " reversed_name+=name return reversed_name class Subject(models.Model): created = models.DateTimeField(auto_now_add=True) name = models.CharField(max_length=200, unique=True) works = models.ManyToManyField("Work", related_name="subjects") class Meta: ordering = ['name'] def __unicode__(self): return self.name class Edition(models.Model): created = models.DateTimeField(auto_now_add=True) title = models.CharField(max_length=1000) publisher_name = models.ForeignKey("PublisherName", related_name="editions", null=True) publication_date = models.CharField(max_length=50, null=True, blank=True, db_index=True,) work = models.ForeignKey("Work", related_name="editions", null=True) cover_image = models.URLField(null=True, blank=True) unglued = models.BooleanField(blank=True) def __unicode__(self): if self.isbn_13: return "%s (ISBN %s) %s" % (self.title, self.isbn_13, self.publisher) if self.oclc: return "%s (OCLC %s) %s" % (self.title, self.oclc, self.publisher) if self.googlebooks_id: return "%s (GOOG %s) %s" % (self.title, self.googlebooks_id, self.publisher) else: return "%s (GLUE %s) %s" % (self.title, self.id, self.publisher) def cover_image_small(self): if self.cover_image: return self.cover_image elif self.googlebooks_id: return "https://encrypted.google.com/books?id=%s&printsec=frontcover&img=1&zoom=5" % self.googlebooks_id else: return '' def cover_image_thumbnail(self): if self.cover_image: return self.cover_image elif self.googlebooks_id: return "https://encrypted.google.com/books?id=%s&printsec=frontcover&img=1&zoom=1" % self.googlebooks_id else: return '' @property def publisher(self): if self.publisher_name: return self.publisher_name.name return '' @property def isbn_10(self): return regluit.core.isbn.convert_13_to_10(self.isbn_13) def id_for(self,type): if not self.pk: return '' try: return self.identifiers.filter(type=type)[0].value except IndexError: return '' @property def isbn_13(self): return self.id_for('isbn') @property def googlebooks_id(self): return self.id_for('goog') @property def librarything_id(self): return self.id_for('thng') @property def oclc(self): return self.id_for('oclc') @property def goodreads_id(self): return self.id_for('gdrd') @property def http_id(self): return self.id_for('http') @staticmethod def get_by_isbn( isbn): if len(isbn)==10: isbn=regluit.core.isbn.convert_10_to_13(isbn) try: return Identifier.objects.get( type='isbn', value=isbn ).edition except Identifier.DoesNotExist: return None def add_author(self, author_name): if author_name: try: author= Author.objects.get(name=author_name) except Author.DoesNotExist: author= Author.objects.create(name=author_name) author.editions.add(self) def set_publisher(self,publisher_name): if publisher_name and publisher_name != '': try: pub_name = PublisherName.objects.get(name=publisher_name) if pub_name.publisher: pub_name = pub_name.publisher.name except PublisherName.DoesNotExist: pub_name = PublisherName.objects.create(name=publisher_name) pub_name.save() self.publisher_name = pub_name self.save() #### following methods for compatibility with marc outputter def downloads(self): return self.ebooks.all() def download_via_url(self): return settings.BASE_URL_SECURE + reverse('download', args=[self.work.id]) def authnames(self): return [auth.last_name_first for auth in self.authors.all()] @property def license(self): try: return self.ebooks.all()[0].rights except: return None @property def funding_info(self): if self.ebooks.all().count()==0: return '' if self.unglued: return 'The book is available as a free download thanks to the generous support of interested readers and organizations, who made donations using the crowd-funding website Unglue.it.' else: if self.ebooks.all()[0].rights in cc.LICENSE_LIST: return 'The book is available as a free download thanks to a Creative Commons license.' else: return 'The book is available as a free download because it is in the Public Domain.' @property def description(self): return self.work.description class Publisher(models.Model): created = models.DateTimeField(auto_now_add=True) name = models.ForeignKey('PublisherName', related_name='key_publisher') url = models.URLField(max_length=1024, null=True, blank=True) logo_url = models.URLField(max_length=1024, null=True, blank=True) description = models.TextField(default='', null=True, blank=True) def __unicode__(self): return self.name.name class PublisherName(models.Model): name = models.CharField(max_length=255, blank=False) publisher = models.ForeignKey('Publisher', related_name='alternate_names', null=True) def __unicode__(self): return self.name def save(self, *args, **kwargs): super(PublisherName, self).save(*args, **kwargs) # Call the "real" save() method. if self.publisher and self != self.publisher.name: #this name is an alias, repoint all editions with this name to the other. for edition in Edition.objects.filter(publisher_name=self): edition.publisher_name = self.publisher.name edition.save() class WasWork(models.Model): work = models.ForeignKey('Work') was = models.IntegerField(unique = True) moved = models.DateTimeField(auto_now_add=True) user = models.ForeignKey(User, null=True) FORMAT_CHOICES = (('pdf','PDF'),( 'epub','EPUB'), ('html','HTML'), ('text','TEXT'), ('mobi','MOBI')) def path_for_file(instance, filename): version = EbookFile.objects.filter(edition = instance.edition, format = instance.format).count() hash = hashlib.md5('%s.%s.%d'%(settings.TWITTER_CONSUMER_SECRET, instance.edition.pk, version)).hexdigest() fn = "ebf/%s.%s"%(hash,instance.format) return fn class EbookFile(models.Model): file = models.FileField(upload_to=path_for_file) format = models.CharField(max_length=25, choices = FORMAT_CHOICES) edition = models.ForeignKey('Edition', related_name='ebook_files') created = models.DateTimeField(auto_now_add=True) asking = models.BooleanField(default=False) def check_file(self): if self.format == 'epub': return test_epub(self.file) return None @property def active(self): try: return Ebook.objects.filter(url=self.file.url)[0].active except: return False class Ebook(models.Model): FORMAT_CHOICES = settings.FORMATS RIGHTS_CHOICES = cc.CHOICES url = models.URLField(max_length=1024) created = models.DateTimeField(auto_now_add=True, db_index=True,) format = models.CharField(max_length=25, choices = FORMAT_CHOICES) provider = models.CharField(max_length=255) download_count = models.IntegerField(default=0) active = models.BooleanField(default=True) # use 'PD-US', 'CC BY', 'CC BY-NC-SA', 'CC BY-NC-ND', 'CC BY-NC', 'CC BY-ND', 'CC BY-SA', 'CC0' rights = models.CharField(max_length=255, null=True, choices = RIGHTS_CHOICES, db_index=True) edition = models.ForeignKey('Edition', related_name='ebooks') user = models.ForeignKey(User, null=True) def set_provider(self): self.provider=Ebook.infer_provider(self.url) return self.provider @property def rights_badge(self): if self.rights is None : return cc.CCLicense.badge('PD-US') return cc.CCLicense.badge(self.rights) @staticmethod def infer_provider( url): if not url: return None # provider derived from url. returns provider value. remember to call save() afterward if re.match('https?://books.google.com/', url): provider='Google Books' elif re.match('https?://www.gutenberg.org/', url): provider='Project Gutenberg' elif re.match('https?://(www\.|)archive.org/', url): provider='Internet Archive' elif url.startswith('http://hdl.handle.net/2027/') or url.startswith('http://babel.hathitrust.org/'): provider='Hathitrust' elif re.match('https?://\w\w\.wikisource\.org/', url): provider='Wikisource' elif re.match('https?://\w\w\.wikibooks\.org/', url): provider='Wikibooks' elif re.match('https://github\.com/\w+/\w+/raw/\w+', url): provider='Github' else: provider=None return provider def increment(self): Ebook.objects.filter(id=self.id).update(download_count = F('download_count') +1) @property def download_url(self): return settings.BASE_URL_SECURE + reverse('download_ebook',args=[self.id]) def is_direct(self): return self.provider!='Google Books' def __unicode__(self): return "%s (%s from %s)" % (self.edition.title, self.format, self.provider) def deactivate(self): self.active=False self.save() def activate(self): self.active=True self.save() class Wishlist(models.Model): created = models.DateTimeField(auto_now_add=True) user = models.OneToOneField(User, related_name='wishlist') works = models.ManyToManyField('Work', related_name='wishlists', through='Wishes') def __unicode__(self): return "%s's Books" % self.user.username def add_work(self, work, source, notify=False): try: w = Wishes.objects.get(wishlist=self,work=work) except: Wishes.objects.create(source=source,wishlist=self,work=work) work.update_num_wishes() # only send notification in case of new wishes # and only when they result from user action, not (e.g.) our tests if notify: wishlist_added.send(sender=self, work=work, supporter=self.user) def remove_work(self, work): w = Wishes.objects.filter(wishlist=self, work=work) if w: w.delete() work.update_num_wishes() def work_source(self, work): w = Wishes.objects.filter(wishlist=self, work=work) if w: return w[0].source else: return '' class Wishes(models.Model): created = models.DateTimeField(auto_now_add=True, db_index=True,) source = models.CharField(max_length=15, blank=True, db_index=True,) wishlist = models.ForeignKey('Wishlist') work = models.ForeignKey('Work', related_name='wishes') class Meta: db_table = 'core_wishlist_works' class Badge(models.Model): name = models.CharField(max_length=72, blank=True) description = models.TextField(default='', null=True) @property def path(self): return '/static/images/%s.png' % self.name def __unicode__(self): return self.name def pledger(): if not pledger.instance: pledger.instance = Badge.objects.get(name='pledger') return pledger.instance pledger.instance=None def pledger2(): if not pledger2.instance: pledger2.instance = Badge.objects.get(name='pledger2') return pledger2.instance pledger2.instance=None ANONYMOUS_AVATAR = '/static/images/header/avatar.png' (NO_AVATAR, GRAVATAR, TWITTER, FACEBOOK, UNGLUEITAR) = (0, 1, 2, 3, 4) class Libpref(models.Model): user = models.OneToOneField(User, related_name='libpref') marc_link_target = models.CharField( max_length=6, default = 'UNGLUE', choices = settings.MARC_PREF_OPTIONS, verbose_name="MARC record link targets" ) class UserProfile(models.Model): created = models.DateTimeField(auto_now_add=True) user = models.OneToOneField(User, related_name='profile') tagline = models.CharField(max_length=140, blank=True) pic_url = models.URLField(blank=True) home_url = models.URLField(blank=True) twitter_id = models.CharField(max_length=15, blank=True) facebook_id = models.PositiveIntegerField(null=True) librarything_id = models.CharField(max_length=31, blank=True) badges = models.ManyToManyField('Badge', related_name='holders') kindle_email = models.EmailField(max_length=254, blank=True) goodreads_user_id = models.CharField(max_length=32, null=True, blank=True) goodreads_user_name = models.CharField(max_length=200, null=True, blank=True) goodreads_auth_token = models.TextField(null=True, blank=True) goodreads_auth_secret = models.TextField(null=True, blank=True) goodreads_user_link = models.CharField(max_length=200, null=True, blank=True) avatar_source = models.PositiveSmallIntegerField(null = True, default = UNGLUEITAR, choices=((NO_AVATAR,'No Avatar, Please'),(GRAVATAR,'Gravatar'),(TWITTER,'Twitter'),(FACEBOOK,'Facebook'),(UNGLUEITAR,'Unglueitar'))) def __unicode__(self): return self.user.username def reset_pledge_badge(self): #count user pledges n_pledges = self.pledge_count if self.badges.exists(): self.badges.remove(pledger()) self.badges.remove(pledger2()) if n_pledges == 1: self.badges.add(pledger()) elif n_pledges > 1: self.badges.add(pledger2()) @property def pledge_count(self): return self.user.transaction_set.exclude(status='NONE').exclude(status='Canceled',reason=None).exclude(anonymous=True).count() @property def account(self): # there should be only one active account per user accounts = self.user.account_set.filter(date_deactivated__isnull=True) if accounts.count()==0: return None else: return accounts[0] @property def old_account(self): accounts = self.user.account_set.filter(date_deactivated__isnull=False).order_by('-date_deactivated') if accounts.count()==0: return None else: return accounts[0] @property def pledges(self): return self.user.transaction_set.filter(status=TRANSACTION_STATUS_ACTIVE) @property def last_transaction(self): from regluit.payment.models import Transaction try: return Transaction.objects.filter(user=self.user).order_by('-date_modified')[0] except IndexError: return None @property def ack_name(self): # use preferences from last transaction, if any last = self.last_transaction if last and last.extra: return last.extra.get('ack_name', self.user.username) else: return self.user.username @property def anon_pref(self): # use preferences from last transaction, if any last = self.last_transaction if last: return last.anonymous else: return None @property def on_ml(self): if "@example.org" in self.user.email: # use @example.org email addresses for testing! return False try: return settings.MAILCHIMP_NEWS_ID in pm.listsForEmail(email_address=self.user.email) except MailChimpException, e: if e.code!=215: # don't log case where user is not on a list logger.error("error getting mailchimp status %s" % (e)) except Exception, e: logger.error("error getting mailchimp status %s" % (e)) return False def ml_subscribe(self, **kwargs): if "@example.org" in self.user.email: # use @example.org email addresses for testing! return True from regluit.core.tasks import ml_subscribe_task ml_subscribe_task.delay(self, **kwargs) def ml_unsubscribe(self): if "@example.org" in self.user.email: # use @example.org email addresses for testing! return True try: return pm.listUnsubscribe(id=settings.MAILCHIMP_NEWS_ID, email_address=self.user.email) except Exception, e: logger.error("error unsubscribing from mailchimp list %s" % (e)) return False def gravatar(self): # construct the url gravatar_url = "https://www.gravatar.com/avatar/" + hashlib.md5(self.user.email.lower()).hexdigest() + "?" gravatar_url += urllib.urlencode({'d':'wavatar', 's':'50'}) return gravatar_url def unglueitar(self): # construct the url gravatar_url = "https://www.gravatar.com/avatar/" + hashlib.md5(self.user.username + '@unglue.it').hexdigest() + "?" gravatar_url += urllib.urlencode({'d':'wavatar', 's':'50'}) return gravatar_url @property def avatar_url(self): if self.avatar_source is None or self.avatar_source is TWITTER: if self.pic_url: return self.pic_url else: return ANONYMOUS_AVATAR elif self.avatar_source == UNGLUEITAR: return self.unglueitar() elif self.avatar_source == GRAVATAR: return self.gravatar() elif self.avatar_source == FACEBOOK and self.facebook_id != None: return 'https://graph.facebook.com/' + str(self.facebook_id) + '/picture' else: return ANONYMOUS_AVATAR @property def social_auths(self): socials= self.user.social_auth.all() auths={} for social in socials: auths[social.provider]=True return auths @property def libraries(self): libs=[] for group in self.user.groups.all(): try: libs.append(group.library) except Library.DoesNotExist: pass return libs class Press(models.Model): url = models.URLField() title = models.CharField(max_length=140) source = models.CharField(max_length=140) date = models.DateField( db_index=True,) language = models.CharField(max_length=20, blank=True) highlight = models.BooleanField(default=False) note = models.CharField(max_length=140, blank=True) # this was causing a circular import problem and we do not seem to be using # anything from regluit.core.signals after this line # from regluit.core import signals from regluit.payment.manager import PaymentManager