tartiflette/app/models/unlistedapps.py

import requests
import json
import datetime
import os
import toml

from .. import db

class UnlistedApp(db.Model):

    id = db.Column(db.Integer, primary_key=True)
    url = db.Column(db.String(100), unique=True, nullable=False)
    name = db.Column(db.String(64), nullable=False)
    owner = db.Column(db.String(64), nullable=False)
    description = db.Column(db.String(256))
    updated_days_ago = db.Column(db.Integer, default=-1)

    def __repr__(self):
        return '<UnlistedApp %r>' % self.name

    def init():
        pass

    def update():

        UnlistedApp.query.delete()

        community = toml.loads(requests.get("https://raw.githubusercontent.com/YunoHost/apps/master/apps.toml").text)

        known_apps = set()
        known_apps = known_apps.union([os.path.basename(app["url"]).lower() for app in community.values() ])

        apps = []

        for i in range(1,8):

            print("Page " + str(i) + " ... ")
            r = requests.get("https://api.github.com/search/repositories?q=_ynh&sort=updated&per_page=100&page="+str(i))
            assert r.status_code == 200, r.text
            j = json.loads(r.text)
            print(str(len(j["items"])) + " items ")
            for item in j["items"]:

                if str(item["size"]) == "0":
                    continue

                if not item["name"].endswith("_ynh"):
                    continue

                if item["name"].lower() in known_apps:
                    continue

                owner = item["owner"]["login"]

                r = requests.head("https://raw.githubusercontent.com/%s/%s/master/manifest.json" % (owner, item["name"]))
                if r.status_code != 200:
                    continue
                r = requests.head("https://raw.githubusercontent.com/%s/%s/master/scripts/install" % (owner, item["name"]))
                if r.status_code != 200:
                    continue

                item["name"] = item["name"].replace("_ynh", "")

                app = UnlistedApp(name=item["name"],
                                  url=item["html_url"],
                                  owner=item["owner"]["login"],
                                  description=item["description"],
                                  updated_days_ago=githubDateToDaysAgo(item["pushed_at"])
                )
                db.session.add(app)

        db.session.commit()


def githubDateToDaysAgo(date):
    now = datetime.datetime.now()
    date = datetime.datetime.strptime(date, "%Y-%m-%dT%H:%M:%SZ")
    return (now - date).days