Микрофиксы

copyexec/copyexec-err.py

@@ -1,36 +1,53 @@
 #!/usr/bin/python
 import sys, os
 
-def header (f):
-  f.write ('/* Generated by python script */\n')
+class SuperFile:
+    def __init__(self, name):
+        self.path = os.path.join(sys.path[0], name + '-test')
+        self.lines = ['/* Generated by python script */']
+    def __enter__(self):
+        return self
+    def __exit__(self, type, value, traceback):
+        with open(self.path, 'w') as f:
+            f.write('\n'.join(self.lines) + '\n')
+    def append(self, line):
+        self.lines.append(line)
 
-errors  = ['LSEEK', 'READ', 'CRC32', 'TYPE', 'DECRYPT', 'MALLOC', 'ALIGN', 'SIZE', 'SHA1', 'MKDIR', 'OPEN', 'FSTAT']
-errors += ['ST_DEV', 'ST_INO', 'READIN', 'STATUS', 'FORK', 'CHDIR', 'CREAT', 'STAT', 'PUBKEY_PERMISSIONS']
+errors  = [
+    'LSEEK', 'READ', 'CRC32', 'TYPE', 'DECRYPT', 'MALLOC', 'ALIGN', 'SIZE',
+    'SHA1', 'MKDIR', 'OPEN', 'FSTAT'
+]
+errors += [
+    'ST_DEV', 'ST_INO', 'READIN', 'STATUS', 'FORK', 'CHDIR', 'CREAT', 'STAT',
+    'PUBKEY_PERMISSIONS'
+]
 errors += ['INVAL']
 
 ########### copyexec-results errors ###########
-errors += ['VOLUME_ID_MISMATCHED', 'HOSTNAME_MISMATCHED', 'DISCONNECT', 'OLD_RESULT']
+errors += [
+    'VOLUME_ID_MISMATCHED', 'HOSTNAME_MISMATCHED', 'DISCONNECT', 'OLD_RESULT'
+]
 
-e = [(s, -(i+1)) for i, s in enumerate (errors)]
-e.append (('UNKNOWN', -999))
+errors = [(s, -(i+1)) for i, s in enumerate (errors)]
+errors.append (('UNKNOWN', -999))
 
-f = open (os.path.join (sys.path[0], 'copyexec-err.h'), 'w')
-header (f)
-for s, i in e:
-  f.write ('#define COPYEXEC_ERR_' + s + ' (' + str (i) + ')\n')
-f.write ('char *copyexec_strerror (int err);\n')
-f.close ()
-f = open (os.path.join (sys.path[0], 'copyexec-err.c'), 'w')
-header (f)
-f.write ('#include <stdio.h>\n')
-f.write ('#include <assert.h>\n')
-f.write ('#include "copyexec-err.h"\n')
-f.write ('char *copyexec_strerror (int err) {\n')
-f.write ('  static char buf[64];')
-f.write ('  switch (err) {\n')
-for s, i in e:
-  f.write ('    case COPYEXEC_ERR_' + s + ': return "COPYEXEC_ERR_' + s + '"; break;\n')
-f.write ('    default: assert (snprintf (buf, sizeof (buf), "COPYEXEC_ERR_UNKNOWN (%d)", err) < sizeof (buf));\n')
-f.write ('             return buf;\n')
-f.write ('  }\n')
-f.write ('}\n')
+with SuperFile('copyexec-err.h') as f:
+     # not using `% tuple_ for tuple_ in errors` just for clarity
+    for error_name, error_code in errors:
+        f.append('#define COPYEXEC_ERR_%s (%d)' % (error_name, error_code))
+    f.append('char *copyexec_strerror (int err);')
+
+with SuperFile('copyexec-err.c') as f:
+    f.append('#include <stdio.h>')
+    f.append('#include <assert.h>')
+    f.append('#include "copyexec-err.h"')
+    f.append('char *copyexec_strerror (int err) {')
+    f.append('  static char buf[64];')
+    f.append('  switch (err) {')
+    for error_name, error_code in errors:
+        f.append('    case COPYEXEC_ERR_%s: return "COPYEXEC_ERR_%s"; break;'
+                 % (error_name, error_name))
+    f.append('    default: assert (snprintf (buf, sizeof (buf), "COPYEXEC_ERR_UNKNOWN (%d)", err) < sizeof (buf));')
+    f.append('             return buf;')
+    f.append('  }')
+    f.append('}')