117 lines
4.0 KiB
Python
117 lines
4.0 KiB
Python
import gi
|
|
import requests
|
|
import os
|
|
import sys
|
|
import inspect
|
|
from gi.repository import Gio
|
|
gi.require_version('AppStreamGlib', '1.0')
|
|
from gi.repository import AppStreamGlib
|
|
from lists import alwaysAccept, alwaysDeny, badLicenses, badCategories, nonFreeAssets, nonFreeNetworkServices
|
|
import itertools
|
|
import argparse
|
|
|
|
#Workaround to get the urls because app.get_urls() doesn't work :|
|
|
def get_urls(app):
|
|
kinds = [AppStreamGlib.UrlKind(kind) for kind in range(11)]
|
|
urls = [(app.get_url_item(kind),kind.value_nick) for kind in kinds]
|
|
return list(filter(lambda a: a[0] is not None, urls))
|
|
|
|
def acceptedGame(app):
|
|
#return 'Game' in app.get_categories()
|
|
if app.get_id() in alwaysAccept:
|
|
return True
|
|
if app.get_id() in alwaysDeny:
|
|
return False
|
|
|
|
return app.get_project_license() and \
|
|
not [x for x in badLicenses if x in app.get_project_license()] and \
|
|
'Game' in app.get_categories() and \
|
|
not [x for x in badCategories if x in app.get_categories()]
|
|
|
|
def getScreenshots(app):
|
|
return [images.get_source() for images in app.get_screenshots()]
|
|
|
|
currentdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
|
|
parentdir = os.path.dirname(currentdir)
|
|
sys.path.insert(0,parentdir)
|
|
|
|
from app.models import *
|
|
from app.utils import make_flask_login_password
|
|
from app.utils.image import get_image_size
|
|
from app.utils import randomString
|
|
|
|
parser = argparse.ArgumentParser(description='Generate HTML files from an Appimage and mako templates.')
|
|
parser.add_argument("-p", '--appstream_path', help='specify the path of the appstream file')
|
|
parser.add_argument("-i", '--icons_path', help='specify the path of the icons')
|
|
parser.add_argument("-m", '--template_path', help='specify the path of the template')
|
|
args = parser.parse_args()
|
|
|
|
store = AppStreamGlib.Store()
|
|
file = Gio.File.new_for_path(args.appstream_path)
|
|
AppStreamGlib.Store.from_file(store, file, args.icons_path, None)
|
|
apps = list(filter(acceptedGame, store.get_apps()))
|
|
session=db.session
|
|
licenses = { x.name : x for x in License.query.all() }
|
|
tags = { x.name : x for x in Tag.query.all() }
|
|
admin_user = User.query.filter_by(username="appstream").first()
|
|
if not admin_user:
|
|
admin_user = User("appstream")
|
|
admin_user.is_active = True
|
|
admin_user.password = make_flask_login_password("appstream")
|
|
admin_user.github_username = "appstream"
|
|
admin_user.forums_username = "appstream"
|
|
admin_user.rank = UserRank.ADMIN
|
|
session.add(admin_user)
|
|
|
|
|
|
for app in apps:
|
|
screenshots = getScreenshots(app)
|
|
urls = get_urls(app)
|
|
filename = app.get_name().replace(':', '').replace('/','') + ".html"
|
|
print("APPLICATION: ",app.get_name())
|
|
game1 = Package()
|
|
game1.state = PackageState.APPROVED
|
|
game1.name = app.get_id().replace('.','_')
|
|
game1.title = app.get_name()
|
|
game1.type = PackageType.GAME
|
|
game1.license = licenses["MIT"]
|
|
#game1.license = licenses[app.get_project_license()]
|
|
game1.media_license = licenses["MIT"]
|
|
game1.author = admin_user
|
|
game1.tags.append(tags["pvp"])
|
|
game1.tags.append(tags["survival"])
|
|
game1.tags.append(tags["multiplayer"])
|
|
for url,t in urls:
|
|
if t == "bugtracker":
|
|
game1.issueTracker = url
|
|
elif t == "homepage":
|
|
game1.repo = url
|
|
|
|
game1.forums = 12835
|
|
game1.short_desc = app.get_comment()
|
|
game1.desc = app.get_description()
|
|
session.add(game1)
|
|
|
|
for i,screenshot in enumerate(screenshots):
|
|
ss = PackageScreenshot()
|
|
ss.package = game1
|
|
ss.title = "Untitled"
|
|
url = screenshot.get_url()
|
|
r = requests.get(url)
|
|
filename = randomString(10) + "." + "png"
|
|
filepath = os.path.join("/var/cdb/uploads", filename)
|
|
with open(filepath,"wb") as f:
|
|
f.write(r.content)
|
|
|
|
ss.url = "uploads/" + filename
|
|
ss.approved = True
|
|
ss.order = i
|
|
ss.width, ss.height = get_image_size(filepath)
|
|
session.add(ss)
|
|
session.commit()
|
|
game1.cover_image = ss
|
|
session.commit()
|
|
|
|
session.commit()
|
|
|