From 6b1b3ac96bcc8ef61c7beb656d631d3446682a90 Mon Sep 17 00:00:00 2001
From: Armen <armen138.pm.me>
Date: Tue, 22 Feb 2022 00:14:20 -0500
Subject: [PATCH] Add flatpak install url to imports

---
 app/tasks/appstreamtasks.py | 23 ++++++++++++-----------
 1 file changed, 12 insertions(+), 11 deletions(-)

diff --git a/app/tasks/appstreamtasks.py b/app/tasks/appstreamtasks.py
index 0391e90..f848ead 100644
--- a/app/tasks/appstreamtasks.py
+++ b/app/tasks/appstreamtasks.py
@@ -11,6 +11,7 @@ import sys
 import inspect
 import shutil
 import urllib.request
+from sqlalchemy.sql import func
 from gi.repository import Gio
 gi.require_version('AppStreamGlib', '1.0')
 from gi.repository import AppStreamGlib
@@ -56,17 +57,6 @@ def importFromFlathub():
     tags = { x.name : x for x in Tag.query.all() }
     admin_user = User.query.filter_by(username="AppStreamBot").first()
 
-    # for ss in PackageScreenshot.query.all():
-    #     if ss.package.cover_image == ss:
-    #         ss.package.cover_image = None
-    #     session.commit()
-    #     session.delete(ss)
-    #     session.commit()
-
-    # for game in Package.query.all():
-    #     session.delete(game)
-    #     session.commit()
-
     if not admin_user:
         admin_user = User("AppStreamBot")
         admin_user.is_active = True
@@ -123,6 +113,17 @@ def importFromFlathub():
             game1.desc = app.get_description()
             session.add(game1)
 
+            install_url = f"https://dl.flathub.org/repo/appstream/{app.get_id()}.flatpakref"
+            release = PackageRelease()        
+            release.package = game1
+            release.title = "Flathub Install"
+            release.url = install_url
+            release.approved = True
+            release.downloads = 0
+            release.releaseDate = func.now()
+
+            session.add(release)
+
             for screenshot in screenshots:
                 counter = 1
                 url = screenshot.get_url()