import random import time from urllib.parse import urlparse from flask import g from sqlalchemy import Column, FetchedValue, ForeignKey from sqlalchemy.orm import deferred, relationship from sqlalchemy.sql.sqltypes import * from files.classes import Base from files.helpers.config.const import * from files.helpers.config.awards import * from files.helpers.slurs_and_profanities import * from files.helpers.lazy import lazy from files.helpers.regex import * from files.helpers.sorting_and_time import make_age_string from .comment import normalize_urls_runtime, add_options, get_award_classes from .polls import * from .sub import * from .subscriptions import * from .saves import SaveRelationship class Post(Base): __tablename__ = "posts" id = Column(Integer, primary_key=True) author_id = Column(Integer, ForeignKey("users.id")) edited_utc = Column(Integer, default=0) created_utc = Column(Integer) thumburl = Column(String) posterurl = Column(String) is_banned = Column(Boolean, default=False) bannedfor = Column(String) chuddedfor = Column(String) ghost = Column(Boolean, default=False) views = Column(Integer, default=0) deleted_utc = Column(Integer, default=0) distinguish_level = Column(Integer, default=0) stickied = Column(String) stickied_utc = Column(Integer) hole_pinned = Column(String) sub = Column(String, ForeignKey("subs.name")) is_pinned = Column(Boolean, default=False) private = Column(Boolean, default=False) comment_count = Column(Integer, default=0) is_approved = Column(Integer, ForeignKey("users.id")) over_18 = Column(Boolean, default=False) is_bot = Column(Boolean, default=False) upvotes = Column(Integer, default=1) downvotes = Column(Integer, default=0) realupvotes = Column(Integer, default=1) app_id = Column(Integer, ForeignKey("oauth_apps.id")) title = Column(String) title_html = Column(String) url = Column(String) body = Column(String) body_html = Column(String) flair = Column(String) ban_reason = Column(String) embed = Column(String) new = Column(Boolean) notify = Column(Boolean) chudded = Column(Boolean, default=False) rainbowed = Column(Boolean, default=False) golden = Column(Boolean, default=False) queened = Column(Boolean, default=False) sharpened = Column(Boolean, default=False) ping_cost = Column(Integer, default=0) bump_utc = Column(Integer) author = relationship("User", primaryjoin="Post.author_id==User.id") oauth_app = relationship("OauthApp") approved_by = relationship("User", uselist=False, primaryjoin="Post.is_approved==User.id") awards = relationship("AwardRelationship", order_by="AwardRelationship.awarded_utc.desc()", back_populates="post") reports = relationship("Report", order_by="Report.created_utc") comments = relationship("Comment", primaryjoin="Comment.parent_post==Post.id", back_populates="post") subr = relationship("Sub", primaryjoin="foreign(Post.sub)==remote(Sub.name)") options = relationship("PostOption", order_by="PostOption.id") def __init__(self, *args, **kwargs): if "created_utc" not in kwargs: kwargs["created_utc"] = int(time.time()) kwargs["bump_utc"] = kwargs["created_utc"] super().__init__(*args, **kwargs) def __repr__(self): return f"<{self.__class__.__name__}(id={self.id})>" @property @lazy def controversial(self): if self.downvotes > 5 and 0.25 < self.upvotes / self.downvotes < 4: return True return False @property @lazy def created_datetime(self): return time.strftime("%d/%B/%Y %H:%M:%S UTC", time.gmtime(self.created_utc)) @property @lazy def age_string(self): return make_age_string(self.created_utc) @property @lazy def edited_string(self): return make_age_string(self.edited_utc) @property @lazy def score(self): return self.upvotes - self.downvotes @property @lazy def fullname(self): return f"p_{self.id}" @property @lazy def id_last_num(self): return str(self.id)[-1] @property @lazy def shortlink(self): link = f"/post/{self.id}" if self.sub: link = f"/h/{self.sub}{link}" if self.sub and self.sub in {'chudrama', 'countryclub', 'highrollerclub'}: output = '-' else: title = self.plaintitle(None).lower() output = title_regex.sub('', title) output = output.split()[:6] output = '-'.join(output) if not output: output = '-' return f"{link}/{output}" @property @lazy def permalink(self): return SITE_FULL + self.shortlink @property @lazy def domain(self): if not self.url: return '' if self.url.startswith('/'): return SITE domain = urlparse(self.url).netloc if domain.startswith("www."): domain = domain.split("www.")[1] return domain.replace("old.reddit.com", "reddit.com") @property @lazy def author_name(self): if self.ghost and not (g.v and self.id == g.v.id): return '👻' return self.author.user_name @property @lazy def is_youtube(self): return self.domain == "youtube.com" and self.embed and self.embed.startswith(' 4 and kind not in {"shit", "fireflies", "gingerbread", "pumpkin"}: return 4 return num @lazy def emoji_award_emojis(self, v, OVER_18_EMOJIS): if g.show_over_18: return [x.note for x in self.awards if x.kind == "emoji"][:4] return [x.note for x in self.awards if x.kind == "emoji" and x.note not in OVER_18_EMOJIS][:4] @lazy def realurl(self, v): url = self.url if not url: return '' url = normalize_urls_runtime(url, v) if url.startswith('/'): return SITE_FULL + url return url @lazy def total_bet_voted(self, v): if "closed" in self.body.lower(): return True if v: for o in self.options: if o.exclusive == 3: return True if o.exclusive == 2 and o.voted(v): return True return False @lazy def total_poll_voted(self, v): if v: if v.id == self.author_id: return True for o in self.options: if o.voted(v): return True return False @lazy def realbody(self, v): if self.deleted_utc != 0 and not (v and (v.admin_level >= PERMS['POST_COMMENT_MODERATION'] or v.id == self.author.id)): return "[Deleted by user]" if self.is_banned and not (v and v.admin_level >= PERMS['POST_COMMENT_MODERATION']) and not (v and v.id == self.author.id): return "" body = self.body_html or "" body = add_options(self, body, v) if self.sub != 'chudrama': body = censor_slurs_profanities(body, v) body = normalize_urls_runtime(body, v) return body @lazy def plainbody(self, v): if self.deleted_utc != 0 and not (v and (v.admin_level >= PERMS['POST_COMMENT_MODERATION'] or v.id == self.author.id)): return "[Deleted by user]" if self.is_banned and not (v and v.admin_level >= PERMS['POST_COMMENT_MODERATION']) and not (v and v.id == self.author.id): return "" body = self.body if not body: return "" if self.sub != 'chudrama': body = censor_slurs_profanities(body, v, True) body = normalize_urls_runtime(body, v) return body @lazy def realtitle(self, v): title = self.title_html if self.sub != 'chudrama': title = censor_slurs_profanities(title, v) return title @lazy def plaintitle(self, v): title = self.title if self.sub != 'chudrama': title = censor_slurs_profanities(title, v, True) return title @property @lazy def is_video(self): return self.url and any((str(self.url).lower().split('?')[0].endswith(f'.{x}') for x in VIDEO_FORMATS)) and is_safe_url(self.url) @property @lazy def is_audio(self): return self.url and any((str(self.url).lower().split('?')[0].endswith(f'.{x}') for x in AUDIO_FORMATS)) and is_safe_url(self.url) @property @lazy def is_image(self): return self.url and any((str(self.url).lower().split('?')[0].endswith(f'.{x}') for x in IMAGE_FORMATS)) and is_safe_url(self.url) @lazy def filtered_reports(self, v): return [r for r in self.reports if not r.user.shadowbanned or (v and v.id == r.user_id) or (v and v.admin_level)] @lazy def active_reports(self, v): return len(self.filtered_reports(v)) @property @lazy def num_subscribers(self): return g.db.query(Subscription).filter_by(post_id=self.id).count() @property @lazy def num_savers(self): return g.db.query(SaveRelationship).filter_by(post_id=self.id).count() @lazy def award_classes(self, v, title=False): return get_award_classes(self, v, title)