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