hub.sr.ht/hubsrht/blueprints/users.py

from sqlalchemy import or_
from sqlalchemy.sql import operators
from flask import Blueprint, render_template, request, abort
from hubsrht.types import User, Project, Visibility, Event, EventType
from hubsrht.types import SourceRepo, MailingList, Tracker
from srht.flask import paginate_query
from srht.oauth import current_user
from srht.search import search_by

users = Blueprint("users", __name__)

@users.route("/~<username>/")
def summary_GET(username):
    user = User.query.filter(User.username == username).first()
    if not user:
        abort(404)
    projects = (Project.query
            .filter(Project.owner_id == user.id)
            .order_by(Project.updated.desc()))
    events = (Event.query
            .filter(Event.user_id == user.id)
            .order_by(Event.created.desc()))

    if not current_user or current_user.id != user.id:
        # TODO: ACLs
        projects = projects.filter(Project.visibility == Visibility.PUBLIC)
        events = (events
                .join(Project, Event.project_id == Project.id)
                .filter(Project.visibility == Visibility.PUBLIC))
        events = (events
            .outerjoin(SourceRepo, Event.source_repo_id == SourceRepo.id)
            .outerjoin(MailingList, Event.source_repo_id == MailingList.id)
            .outerjoin(Tracker, Event.source_repo_id == Tracker.id)
            .filter(or_(Event.source_repo == None, SourceRepo.visibility == Visibility.PUBLIC),
                or_(Event.mailing_list == None, MailingList.visibility == Visibility.PUBLIC),
                or_(Event.tracker == None, Tracker.visibility == Visibility.PUBLIC)))

    projects = projects.limit(5).all()
    events, pagination = paginate_query(events)

    return render_template("profile.html",
            user=user, projects=projects, EventType=EventType, events=events,
            **pagination)

@users.route("/projects/<owner>/")
def projects_GET(owner):
    if owner.startswith("~"):
        owner = owner[1:]
    owner = User.query.filter(User.username == owner).first()
    if not owner:
        abort(404)
    projects = (Project.query
        .filter(Project.owner_id == owner.id))
    if not current_user or current_user.id != owner.id:
        # TODO: ACLs
        projects = projects.filter(Project.visibility == Visibility.PUBLIC)

    search = request.args.get("search")
    search_error = None
    if search:
        try:
            projects = search_by(projects, search,
                    [Project.name, Project.description],
                    key_fns={"tag": lambda t:
                        Project.tags.any(t, operator=operators.ilike_op)},
                    term_map=lambda t: f"tag:{t[1:]}" if t.startswith("#") else t)
        except ValueError as e:
            search_error = str(e)

    projects, pagination = paginate_query(projects)

    return render_template("projects.html", user=owner, projects=projects,
            search=search, search_error=search_error, **pagination)