From 7b111a604b4560a209216332b1bd526d9258a5a3 Mon Sep 17 00:00:00 2001 From: David Cantrell Date: Fri, 12 Sep 2008 15:05:49 -1000 Subject: [PATCH] Added treeinfo.py module to pylorax. The treeinfo module replaces maketreeinfo.py and currently provides a single function called write(). --- pylorax/__init__.py | 3 +- pylorax/treeinfo.py | 93 +++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 95 insertions(+), 1 deletion(-) create mode 100644 pylorax/treeinfo.py diff --git a/pylorax/__init__.py b/pylorax/__init__.py index bb8999d3..58b6a084 100644 --- a/pylorax/__init__.py +++ b/pylorax/__init__.py @@ -22,9 +22,10 @@ version = (0, 1) -__all__ = ['discinfo'] +__all__ = ['discinfo', 'treeinfo'] import discinfo +import treeinfo def show_version(prog): """show_version(prog) diff --git a/pylorax/treeinfo.py b/pylorax/treeinfo.py new file mode 100644 index 00000000..92aa0e76 --- /dev/null +++ b/pylorax/treeinfo.py @@ -0,0 +1,93 @@ +# +# pylorax treeinfo module +# Install image and tree support data generation tool -- Python module. +# +# Copyright (C) 2007, 2008 Red Hat, Inc. +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see . +# +# Author(s): David Cantrell +# Will Woods +# + +import os +import time +import ConfigParser + +# Write out the .treeinfo file +def write(family=None, version=None, arch=None, outdir=None, variant='', + discnum=None, totaldiscs=None, packagedir='', allDiscs=True): + """write(family=None, version=None, arch=None, outdir=None, + [variant='', discnum=None, totaldiscs=None, packagedir='']) + + Write the .treeinfo file to the specified directory (outdir). + + Required parameters: + family String specifying the family. + version String specifying the version number. + arch String specifying the architecture. + outdir Directory to write .treeinfo to (must exist). + + Optional parameters may be specified: + variant Defaults to an empty string, but can be any string. + discnum Defaults to '1', but you can specify an integer. + totaldiscs Defaults to '1', but you can specify an integer. + packagedir Directory where packages are located. + allDiscs Boolean stating all discs are in one tree (default: True) + + Returns True on success, False on failure. + + """ + + if family is None or arch is None or outdir is None: + return False + + if not os.path.isdir(outdir): + return False + + data = { "timestamp": float(time.time()), + "family": family, + "variant": variant, + "version": version, + "arch": arch, + "discnum": "1", + "totaldiscs": "1", + "packagedir": packagedir, + "outfile": None } + + outfile = "%s/.treeinfo" % (outdir,) + section = 'general' + + c = ConfigParser.ConfigParser() + c.add_section(section) + + if not allDiscs: + if discnum is not None: + data["discnum"] = str(discnum) + + if totaldiscs is not None: + data["totaldiscs"] = str(totaldiscs) + + try: + f = open(outfile, "w") + + for key, value in data.items(): + c.set(section, key, value) + + c.write(f) + f.close() + except: + return False + + return True