[aur-dev] [PATCH 11/17] mkpkglists: Make output files configurable

[Lukas Fleischer]

Instead of writing the output to hardcoded files (relative to the
document root), make the output paths of mkpkglists configurable.

Signed-off-by: Lukas Fleischer <lfleischer at archlinux.org>
---
 conf/config.proto     | 4 ++++
 scripts/mkpkglists.py | 9 +++++----
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/conf/config.proto b/conf/config.proto
index c56141c..e545977 100644
--- a/conf/config.proto
+++ b/conf/config.proto
@@ -63,3 +63,7 @@ max-blob-size = 256000
 db-path = /srv/http/aurweb/aurblup/
 sync-dbs = core extra community multilib testing community-testing
 servers = ftp://mirrors.kernel.org/archlinux/%s/os/x86_64
+
+[mkpkglists]
+packagesfile = /srv/http/aurweb/web/html/packages.gz
+pkgbasefile = /srv/http/aurweb/web/html/pkgbase.gz
diff --git a/scripts/mkpkglists.py b/scripts/mkpkglists.py
index 70cbd13..8a0f2e9 100755
--- a/scripts/mkpkglists.py
+++ b/scripts/mkpkglists.py
@@ -2,11 +2,12 @@
 
 import datetime
 import gzip
-import os
 
+import aurweb.config
 import aurweb.db
 
-docroot = os.path.dirname(os.path.realpath(__file__)) + "/../web/html/"
+packagesfile = aurweb.config.get('mkpkglists', 'packagesfile')
+pkgbasefile = aurweb.config.get('mkpkglists', 'pkgbasefile')
 
 
 def main():
@@ -16,7 +17,7 @@ def main():
     pkglist_header = "# AUR package list, generated on " + datestr
     pkgbaselist_header = "# AUR package base list, generated on " + datestr
 
-    with gzip.open(docroot + "packages.gz", "w") as f:
+    with gzip.open(packagesfile, "w") as f:
         f.write(bytes(pkglist_header + "\n", "UTF-8"))
         cur = conn.execute("SELECT Packages.Name FROM Packages " +
                            "INNER JOIN PackageBases " +
@@ -24,7 +25,7 @@ def main():
                            "WHERE PackageBases.PackagerUID IS NOT NULL")
         f.writelines([bytes(x[0] + "\n", "UTF-8") for x in cur.fetchall()])
 
-    with gzip.open(docroot + "pkgbase.gz", "w") as f:
+    with gzip.open(pkgbasefile, "w") as f:
         f.write(bytes(pkgbaselist_header + "\n", "UTF-8"))
         cur = conn.execute("SELECT Name FROM PackageBases " +
                            "WHERE PackagerUID IS NOT NULL")
-- 
2.10.0