<filename>collipa/models/tweet.py<gh_stars>10-100
# coding: utf-8

import time
from pony import orm
from ._base import db, BaseModel
import collipa.models
from collipa import config
from collipa.helpers import strip_tags, process_content


class Tweet(db.Entity, BaseModel):
    user_id = orm.Required(int)

    content = orm.Required(orm.LongUnicode)

    role = orm.Required(unicode, 10, default='tweet')

    thank_count = orm.Required(int, default=0)
    up_count = orm.Required(int, default=0)
    down_count = orm.Required(int, default=0)
    report_count = orm.Required(int, default=0)
    reply_count = orm.Required(int, default=0)
    collect_count = orm.Required(int, default=0)

    created_at = orm.Required(int, default=int(time.time()))
    updated_at = orm.Required(int, default=int(time.time()))
    active = orm.Required(int, default=int(time.time()))
    has_img = orm.Optional(unicode, 10)

    def __str__(self):
        return self.id

    def __repr__(self):
        return '<Tweet: %s>' % self.id

    @property
    def url(self):
        return '/tweet/%s' % self.id

    def to_simple_dict(self):
        return dict(
            id=self.id,
            content=self.content,
            clean_content=strip_tags(self.content),
            reply_count=self.reply_count,
            has_image=self.has_img == 'true',
            created_at=self.created_at,
            created=self.created,
        )

    def to_dict(self):
        user = collipa.models.User[self.user_id]
        data = self.to_simple_dict()
        data.update(dict(
            author=user.to_simple_dict(),
            images=[img.to_simple_dict() for img in self.images],
        ))

        return data

    def save(self, category='create', user=None):
        now = int(time.time())
        if category == 'create':
            self.created_at = now
            self.author.tweet_count += 1

        if not user:
            user = self.author

        self.content = process_content(self.content)
        self.updated_at = now
        self.active = now

        user.active = now

        return super(Tweet, self).save()

    def delete(self, user=None):
        self.author.tweet_count -= 1
        for th in collipa.models.Thank.select(lambda rv: rv.tweet_id == self.id):
            th.delete()
        for up in collipa.models.Up.select(lambda rv: rv.tweet_id == self.id):
            up.delete()
        for dw in collipa.models.Down.select(lambda rv: rv.tweet_id == self.id):
            dw.delete()
        for rp in collipa.models.Report.select(lambda rv: rv.tweet_id == self.id):
            rp.delete()

        if not user:
            user = self.author
        user.active = int(time.time())

        super(Tweet, self).delete()

    def get_uppers(self, after_date=None, before_date=None):
        if after_date:
            user_ids = orm.select(rv.user_id for rv in collipa.models.Up
                                  if rv.tweet_id == self.id and rv.created_at > after_date)
        elif before_date:
            user_ids = orm.select(rv.user_id for rv in collipa.models.Up
                                  if rv.tweet_id == self.id and rv.created_at < before_date)
        else:
            user_ids = orm.select(rv.user_id for rv in collipa.models.Up if rv.tweet_id == self.id)
        users = []
        if user_ids:
            user_ids = user_ids.order_by(lambda: orm.desc(rv.created_at))

            users = orm.select(rv for rv in collipa.models.User if rv.id in user_ids)
        return users

    def get_thankers(self, after_date=None, before_date=None):
        if after_date:
            user_ids = orm.select(rv.user_id for rv in collipa.models.Thank
                                  if rv.tweet_id == self.id and rv.created_at > after_date)
        elif before_date:
            user_ids = orm.select(rv.user_id for rv in collipa.models.Thank
                                  if rv.tweet_id == self.id and rv.created_at < before_date)
        else:
            user_ids = orm.select(rv.user_id for rv in collipa.models.Thank if rv.tweet_id == self.id)
        users = []
        if user_ids:
            user_ids = user_ids.order_by(lambda: orm.desc(rv.created_at))

            users = orm.select(rv for rv in collipa.models.User if rv.id in user_ids)
        return users

    def put_notifier(self):
        content, users = self.get_compiled_content_mention_users(self.content)
        self.content = content
        for user in users:
            if user.id != self.user_id:
                collipa.models.Notification(tweet_id=self.id, receiver_id=user.id, role='mention').save()
        return self

    @staticmethod
    def get_timeline(page=1, from_id=None, user_id=None, count=config.paged):
        if not from_id:
            if not user_id:
                tweets = (orm.select(rv for rv in Tweet)
                          .order_by(lambda: orm.desc(rv.created_at))
                          [(page - 1) * count: page * count])
            else:
                tweets = (orm.select(rv for rv in Tweet if rv.user_id == user_id)
                          .order_by(lambda: orm.desc(rv.created_at))
                          [(page - 1) * count: page * count])
        else:
            if not user_id:
                tweets = (orm.select(rv for rv in collipa.models.Tweet if rv.id < from_id)
                          .order_by(lambda: orm.desc(rv.created_at))
                          [:count])
            else:
                tweets = (orm.select(rv for rv in collipa.models.Tweet if rv.id < from_id and rv.user_id == user_id)
                          .order_by(lambda: orm.desc(rv.created_at))
                          [:count])
        return tweets

    @property
    def images(self):
        return collipa.models.Image.select(lambda rv: rv.tweet_id == self.id)
