[Pkg-python-debian-commits] r66 trunk: - clean up module namespace

[zack at bononia.it]

------------------------------------------------------------
revno: 66
committer: zack at bononia.it
branch nick: pkg-python-debian.debfile
timestamp: Sat 2007-07-14 11:21:56 +0200
message:
  - clean up module namespace
  - proof of concept test
  - __init__ using convention of TarFile/Gzip/...
modified:
  debian_bundle/debfile.py
-------------- next part --------------
=== modified file 'debian_bundle/debfile.py'
--- a/debian_bundle/debfile.py	2007-07-14 08:59:46 +0000
+++ b/debian_bundle/debfile.py	2007-07-14 09:21:56 +0000
@@ -1,6 +1,8 @@
-import arfile
+#!/usr/bin/python
+
 import gzip
 import tarfile
+from arfile import ArFile, ArError
 
 DATA_PART = 'data.tar.gz'
 CTRL_PART = 'control.tar.gz'
@@ -10,7 +12,7 @@
 CTRL_FILE = 'control'
 MD5_FILE = 'md5sums'
 
-class DebError(arfile.ArError):
+class DebError(ArError):
     pass
 
 
@@ -106,11 +108,11 @@
         md5_file.close()
         return sums
 
-class DebFile(arfile.ArFile):
+class DebFile(ArFile):
 
-    def __init__(self, **args):
-        ArFile.__init__(self, **args)
-        if set(self.getnames()) != set([INFO_PART, CTRL_PART, DATA_PART])
+    def __init__(self, filename=None, mode='r', fileobj=None):
+        ArFile.__init__(self, filename, mode, fileobj)
+        if set(self.getnames()) != set([INFO_PART, CTRL_PART, DATA_PART]):
             raise DebError('unexpected .deb content')
 
         self.__parts = {}
@@ -130,3 +132,7 @@
     def getCtrl(self): return self.__parts[CTRL_PART]
     control = property(getCtrl)
 
+if __name__ == '__main__':
+    import sys
+    deb = DebFile(filename=sys.argv[1])
+