207 lines
5 KiB
Python
207 lines
5 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
|
import os
|
|
import tarfile
|
|
|
|
|
|
|
|
|
|
class ArchDatabase:
|
|
"""Representation of an Arch Linux package database."""
|
|
FILE_EXTENSION = ".db.tar.gz"
|
|
|
|
|
|
def __init__(self, filename):
|
|
"""Construct a new database object and read all containing packages."""
|
|
self.filename = filename
|
|
self.packages = []
|
|
self._parse()
|
|
|
|
|
|
def get_fielname(self):
|
|
"""Get filename of database."""
|
|
return self.filename
|
|
|
|
|
|
def get_name(self):
|
|
"""Get simple name of database."""
|
|
return os.path.basename(self.filename)[0:-len(ArchDatabase.FILE_EXTENSION)]
|
|
|
|
|
|
def get_packages(self):
|
|
"""Get list of packages contained in the database."""
|
|
return self.packages
|
|
|
|
|
|
def _parse(self):
|
|
"""Read and parse database content."""
|
|
with tarfile.open(name=self.filename) as tf:
|
|
for f in tf:
|
|
if f.isfile() and os.path.basename(f.name) == "desc":
|
|
df = tf.extractfile(f)
|
|
s = df.read().decode('utf-8')
|
|
|
|
d = dict(item.split("\n")[:2] for item in s.split("\n\n")[:-1])
|
|
if "%NAME%" in d:
|
|
archPackage = ArchPackage(d["%NAME%"])
|
|
if "%FILENAME%" in d:
|
|
archPackage.set_filename(d["%FILENAME%"])
|
|
if "%VERSION%" in d:
|
|
archPackage.set_version(d["%VERSION%"])
|
|
if "%DESC%" in d:
|
|
archPackage.set_desc(d["%DESC%"])
|
|
if "%CSIZE%" in d:
|
|
archPackage.set_csize(d["%CSIZE%"])
|
|
if "%ISIZE%" in d:
|
|
archPackage.set_isize(d["%ISIZE%"])
|
|
if "%URL%" in d:
|
|
archPackage.set_url(d["%URL%"])
|
|
if "%LICENSE%" in d:
|
|
archPackage.set_license(d["%LICENSE%"])
|
|
if "%ARCH%" in d:
|
|
archPackage.set_arch(d["%ARCH%"])
|
|
if "%BUILDDATE%" in d:
|
|
archPackage.set_builddate(d["%BUILDDATE%"])
|
|
if "%PACKAGER%" in d:
|
|
archPackage.set_packager(d["%PACKAGER%"])
|
|
self.packages.append(archPackage)
|
|
|
|
|
|
def find_databases(directory):
|
|
"""Find all database files in a directory, including subdirectories."""
|
|
databases = []
|
|
for root, dirs, files in os.walk(directory):
|
|
for file in files:
|
|
if file.endswith(ArchDatabase.FILE_EXTENSION):
|
|
archDatabase = ArchDatabase(os.path.join(root, file))
|
|
databases.append(archDatabase)
|
|
return databases
|
|
|
|
|
|
|
|
|
|
class ArchPackage:
|
|
"""Representation of an Arch Linux package."""
|
|
|
|
|
|
def __init__(self, name):
|
|
"""Construct a new package."""
|
|
self.name = name
|
|
self.filename = None
|
|
self.version = None
|
|
self.desc = None
|
|
self.csize = None
|
|
self.isize = None
|
|
self.url = None
|
|
self.license = None
|
|
self.arch = None
|
|
self.builddate = None
|
|
self.packager = None
|
|
|
|
|
|
def get_name(self):
|
|
"""Get the package name."""
|
|
return self.name
|
|
|
|
|
|
def set_filename(self, filename):
|
|
"""Set package filename."""
|
|
self.filename = filename
|
|
|
|
|
|
def get_filename(self):
|
|
"""Get package filename."""
|
|
return self.filename
|
|
|
|
|
|
def set_version(self, version):
|
|
"""Set version."""
|
|
self.version = version
|
|
|
|
|
|
def get_version(self):
|
|
"""Get version."""
|
|
return self.version
|
|
|
|
|
|
def set_desc(self, desc):
|
|
"""Set description."""
|
|
self.desc = desc
|
|
|
|
|
|
def get_desc(self):
|
|
"""Get description."""
|
|
return self.desc
|
|
|
|
|
|
def set_csize(self, csize):
|
|
"""Set csize."""
|
|
self.csize = csize
|
|
|
|
|
|
def get_csize(self):
|
|
"""Get csize."""
|
|
return self.csize
|
|
|
|
|
|
def set_isize(self, isize):
|
|
"""Set isize."""
|
|
self.isize = isize
|
|
|
|
|
|
def get_isize(self):
|
|
"""Get isize."""
|
|
return self.isize
|
|
|
|
|
|
def set_url(self, url):
|
|
"""Set upstream URL."""
|
|
self.url = url
|
|
|
|
|
|
def get_url(self):
|
|
"""Get upstream URL."""
|
|
return self.url
|
|
|
|
|
|
def set_license(self, license):
|
|
"""Set license."""
|
|
self.license = license
|
|
|
|
|
|
def get_license(self):
|
|
"""Get license."""
|
|
return self.license
|
|
|
|
|
|
def set_arch(self, arch):
|
|
"""Set architecture."""
|
|
self.arch = arch
|
|
|
|
|
|
def get_arch(self):
|
|
"""Get architecture."""
|
|
return self.arch
|
|
|
|
|
|
def set_builddate(self, builddate):
|
|
"""Set build date."""
|
|
self.builddate = builddate
|
|
|
|
|
|
def get_builddate(self):
|
|
"""Get build date."""
|
|
return self.builddate
|
|
|
|
|
|
def set_packager(self, packager):
|
|
"""Set name of packager."""
|
|
self.packager = packager
|
|
|
|
|
|
def get_packager(self):
|
|
"""Get name of packager."""
|
|
return self.packager
|