diff --git a/pylorax/__init__.py b/pylorax/__init__.py index 061db1bf..bb8999d3 100644 --- a/pylorax/__init__.py +++ b/pylorax/__init__.py @@ -22,7 +22,18 @@ version = (0, 1) +__all__ = ['discinfo'] + +import discinfo + def show_version(prog): + """show_version(prog) + + Display program name (prog) and version number. If prog is an empty + string or None, use the value 'pylorax'. + + """ + if prog is None or prog == '': prog = 'pylorax' diff --git a/pylorax/discinfo.py b/pylorax/discinfo.py new file mode 100644 index 00000000..f9aa7e28 --- /dev/null +++ b/pylorax/discinfo.py @@ -0,0 +1,62 @@ +# +# pylorax discinfo module +# Install image and tree support data generation tool -- Python module. +# +# Copyright (C) 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 +# + +import os +import time + +# Write out the .discinfo file +def write(release=None, arch=None, outdir=None, disc=None): + """write(release=None, arch=None, outdir=None, [disc=None]) + + Write the .discinfo file to the specified directory (outdir). + The release string is specified as release and the architecture + is specified as arch. If disc is specified, it will be written + as the disc number, otherwise 0 is written for the disc number. + + The release, arch, and outdir parameters are all required. + + """ + + if release is None or arch is None or outdir is None: + return False + + if not os.path.isdir(outdir): + return False + + outfile = "%s/.discinfo" % (outdir,) + + try: + f = open(outfile, "w") + f.write("%s\n", time.time()) + f.write("%s\n", release) + f.write("%s\n", arch) + + if disc is not None: + f.write("%d\n", disc) + else: + f.write("0\n") + + f.close() + except: + return False + + return True