Add rss.py
This commit is contained in:
Louis Abel
parent
19b0a1206f
commit
f80f1ddabf
GPG Key ID:
3331F061D1D9990E
221
files/usr/libexec/rocky/rss.py
Normal file
221
files/usr/libexec/rocky/rss.py
Normal file
@ -0,0 +1,221 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*-:python; coding:utf-8; -*-
|
||||
# author: Louis Abel <label@rockylinux.org>
|
||||
# modified version of repo-rss from yum utils
|
||||
|
||||
import sys
|
||||
import os
|
||||
import argparse
|
||||
import time
|
||||
import binascii
|
||||
import dnf
|
||||
import dnf.exceptions
|
||||
#from dnf.comps import Comps
|
||||
import libxml2
|
||||
|
||||
def to_unicode(string: str) -> str:
|
||||
"""
|
||||
Convert to unicode
|
||||
"""
|
||||
if isinstance(string, bytes):
|
||||
return string.decode('utf8')
|
||||
if isinstance(string, str):
|
||||
return string
|
||||
return str(string)
|
||||
|
||||
class DnfQuiet(dnf.Base):
|
||||
"""
|
||||
DNF object
|
||||
"""
|
||||
def __init__(self):
|
||||
dnf.Base.__init__(self)
|
||||
|
||||
def get_recent(self, days=1):
|
||||
"""
|
||||
Return most recent packages from dnf sack
|
||||
"""
|
||||
recent = []
|
||||
now = time.time()
|
||||
recentlimit = now-(days*86400)
|
||||
ftimehash = {}
|
||||
if self.conf.showdupesfromrepos:
|
||||
available = self.sack.query().available().filter()
|
||||
else:
|
||||
available = self.sack.query().available().filter(latest_per_arch=1)
|
||||
|
||||
for package in available:
|
||||
ftime = int(package.buildtime)
|
||||
if ftime > recentlimit:
|
||||
if ftime not in ftimehash:
|
||||
ftimehash[ftime] = [package]
|
||||
else:
|
||||
ftimehash[ftime].append(package)
|
||||
|
||||
for sometime in ftimehash.keys():
|
||||
for package in ftimehash[sometime]:
|
||||
recent.append(package)
|
||||
|
||||
return recent
|
||||
|
||||
class RepoRSS:
|
||||
def __init__(self, filename='repo-rss.xml'):
|
||||
self.description = 'Repository RSS'
|
||||
self.link = 'http://dnf.baseurl.org'
|
||||
self.title = 'Recent Packages'
|
||||
self.do_file(filename)
|
||||
self.do_doc()
|
||||
|
||||
def do_doc(self):
|
||||
self.doc = libxml2.newDoc('1.0')
|
||||
self.xmlescape = self.doc.encodeEntitiesReentrant
|
||||
rss = self.doc.newChild(None, 'rss', None)
|
||||
rss.setProp('version', '2.0')
|
||||
self.rssnode = rss.newChild(None, 'channel', None)
|
||||
|
||||
def do_file(self, filename):
|
||||
if filename[0] != '/':
|
||||
cwd = os.getcwd()
|
||||
self.filename = os.path.join(cwd, filename)
|
||||
else:
|
||||
self.filename = filename
|
||||
|
||||
try:
|
||||
self.file_open = open(self.filename, 'w+')
|
||||
except IOError as exc:
|
||||
print(f'Error opening file {self.filename}: {exc}', file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
def rsspackage(self, package):
|
||||
rfc822_format = "%a, %d %b %Y %X GMT"
|
||||
changelog_format = "%a, %d %b %Y GMT"
|
||||
package_hex = binascii.hexlify(package.chksum[1]).decode()
|
||||
item = self.rssnode.newChild(None, 'item', None)
|
||||
title = self.xmlescape(str(package))
|
||||
description = package.description
|
||||
item.newChild(None, 'title', title)
|
||||
date = time.gmtime(float(package.buildtime))
|
||||
item.newChild(None, 'pubDate', time.strftime(rfc822_format, date))
|
||||
# pylint: disable=line-too-long
|
||||
item.newChild(None, 'guid', package_hex).setProp("isPermaLink", "false")
|
||||
link = package.remote_location()
|
||||
item.newChild(None, 'link', self.xmlescape(link))
|
||||
changelog = ''
|
||||
count = 0
|
||||
if package.changelogs is not None:
|
||||
changelog_list = package.changelogs
|
||||
else:
|
||||
changelog_list = []
|
||||
for meta in changelog_list:
|
||||
count += 1
|
||||
if count > 3:
|
||||
changelog += '...'
|
||||
break
|
||||
date = meta['timestamp'].strftime(changelog_format)
|
||||
author = meta['author']
|
||||
desc = meta['text']
|
||||
changelog += f'{date} - {author}\n{desc}\n\n'
|
||||
# pylint: disable=line-too-long,consider-using-f-string
|
||||
description = '<p><strong>{}</strong> - {}</p>\n\n'.format(self.xmlescape(package.name), self.xmlescape(package.summary))
|
||||
description += '<p>%s</p>\n\n<p><strong>Change Log:</strong></p>\n\n' % self.xmlescape(description.replace("\n", "<br />\n"))
|
||||
description += self.xmlescape('<pre>%s</pre>' % self.xmlescape(changelog))
|
||||
item.newChild(None, 'description', description)
|
||||
return item
|
||||
|
||||
def start_rss(self):
|
||||
"""return string representation of rss preamble"""
|
||||
rfc822_format = "%a, %d %b %Y %X GMT"
|
||||
now = time.strftime(rfc822_format, time.gmtime())
|
||||
rssheader = f"""<?xml version="1.0" encoding="utf-8"?>
|
||||
<rss version="2.0">
|
||||
<channel>
|
||||
<title>{self.title}</title>
|
||||
<link>{self.link}</link>
|
||||
<description>{self.description}</description>
|
||||
<pubDate>{now}</pubDate>
|
||||
<generator>DNF</generator>
|
||||
"""
|
||||
|
||||
self.file_open.write(rssheader)
|
||||
|
||||
def do_package(self, package):
|
||||
item = self.rsspackage(package)
|
||||
self.file_open.write(item.serialize("utf-8", 1))
|
||||
item.unlinkNode()
|
||||
item.freeNode()
|
||||
del item
|
||||
|
||||
def close_rss(self):
|
||||
end="\n </channel>\n</rss>\n"
|
||||
self.file_open.write(end)
|
||||
self.file_open.close()
|
||||
del self.file_open
|
||||
self.doc.freeDoc()
|
||||
del self.doc
|
||||
|
||||
def make_rss_feed(filename, title, link, description, recent, dnfobj):
|
||||
rssobj = RepoRSS(filename)
|
||||
rssobj.title = title
|
||||
rssobj.link = link
|
||||
rssobj.description = description
|
||||
rssobj.start_rss()
|
||||
if len(recent) > 0:
|
||||
for package in recent:
|
||||
rssobj.do_package(package)
|
||||
rssobj.close_rss()
|
||||
|
||||
def main(options):
|
||||
days = options.days
|
||||
repoids = options.repoids
|
||||
dnfobj = DnfQuiet()
|
||||
if options.config:
|
||||
dnfobj.conf.read(filename=options.config)
|
||||
|
||||
if os.geteuid() != 0 or options.tempcache:
|
||||
cachedir = dnfobj.conf.cachedir
|
||||
if cachedir is None:
|
||||
print('Error: Could not make cachedir')
|
||||
sys.exit(50)
|
||||
dnfobj.conf.cachedir = cachedir
|
||||
|
||||
try:
|
||||
dnfobj.read_all_repos()
|
||||
except:
|
||||
print('Could not read repos', file=sys.stderr)
|
||||
sys.exit(1)
|
||||
|
||||
if len(repoids) > 0:
|
||||
for repo in dnfobj.repos:
|
||||
repoobj = dnfobj.repos[repo]
|
||||
if repo not in repoids:
|
||||
repoobj.disable()
|
||||
else:
|
||||
repoobj.enable()
|
||||
repoobj.load_metadata_other = True
|
||||
|
||||
print('Getting repo data')
|
||||
try:
|
||||
dnfobj.fill_sack()
|
||||
except:
|
||||
print('repo data failure')
|
||||
sys.exit(1)
|
||||
|
||||
sack_query = dnfobj.sack.query().available()
|
||||
recent = sack_query.filter(latest_per_arch=1)
|
||||
sorted_recents = sorted(set(recent.run()), key=lambda pkg: pkg.buildtime)
|
||||
sorted_recents.reverse()
|
||||
make_rss_feed(options.filename, options.title, options.link, options.description, sorted_recents, dnfobj)
|
||||
|
||||
if __name__ == "__main__":
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument('--filename', type=str, default='repo-rss.xml')
|
||||
parser.add_argument('--link', type=str, default='http://yum.baseurl.org')
|
||||
parser.add_argument('--title', type=str, default='RSS Repository - Recent Packages')
|
||||
parser.add_argument('--description', type=str, default='Most recent packages in Repositories')
|
||||
parser.add_argument('--days', type=int, default=7)
|
||||
parser.add_argument('--tempcache', action='store_true')
|
||||
parser.add_argument('--arches', action='append', default=[])
|
||||
parser.add_argument('--config', type=str, default='')
|
||||
parser.add_argument('repoids', metavar='N', type=str, nargs='+')
|
||||
results = parser.parse_args()
|
||||
|
||||
main(results)
|
||||
@ -2,6 +2,9 @@
|
||||
- name: Ensure docroot actually exists
|
||||
ansible.builtin.file:
|
||||
path: "{{ feeds_output_dir }}"
|
||||
owner: '10004'
|
||||
group: '10005'
|
||||
mode: '0755'
|
||||
state: directory
|
||||
|
||||
- name: Generate repo file for RSS feed data
|
||||
@ -19,4 +22,12 @@
|
||||
owner: root
|
||||
group: root
|
||||
mode: '0755'
|
||||
|
||||
- name: Deploy the RSS feeder script itself
|
||||
ansible.builtin.copy:
|
||||
src: "usr/libexec/rocky/rss.py"
|
||||
dest: "/usr/libexec/rocky/rss.py"
|
||||
owner: root
|
||||
group: root
|
||||
mode: '0755'
|
||||
...
|
||||
|
||||
Loading…
Reference in New Issue
Block a user