Merge pull request #4 from vuolter/0.4.10

vuolter HEAD

1 files changed, 11 insertions, 10 deletions

diff --git a/pyload/remote/socketbackend/create_ttypes.py b/pyload/remote/socketbackend/create_ttypes.py
index 9b001f1bf..00752dc6b 100644
--- a/pyload/remote/socketbackend/create_ttypes.py
+++ b/pyload/remote/socketbackend/create_ttypes.py
@@ -1,5 +1,7 @@
 # -*- coding: utf-8 -*-
 
+from __future__ import with_statement
+
 import inspect
 import os
 import platform
@@ -12,7 +14,6 @@ from pyload.remote.thriftbackend.thriftgen.pyload.Pyload import Iface
 
 
 def main():
-
     enums = []
     classes = []
 
@@ -21,19 +22,19 @@ def main():
     for name in dir(ttypes):
         klass = getattr(ttypes, name)
 
-        if name in ("TBase", "TExceptionBase") or name.startswith("_") or not (issubclass(klass, ttypes.TBase) or issubclass(klass, ttypes.TExceptionBase)):
+        if name in ("TBase", "TExceptionBase") or name.startswith("_") or not (
+            issubclass(klass, ttypes.TBase) or issubclass(klass, ttypes.TExceptionBase)):
             continue
 
         if hasattr(klass, "thrift_spec"):
-           classes.append(klass)
+            classes.append(klass)
         else:
             enums.append(klass)
 
 
-    f = open(os.path.join(pypath, "pyload", "api", "types.py"), "wb")
-
-    f.write(
-        """# -*- coding: utf-8 -*-
+    with open(os.path.join(pypath, "pyload", "api", "types.py"), "wb") as f:
+        f.write(
+"""# -*- coding: utf-8 -*-
 # Autogenerated by pyload
 # DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
 
@@ -43,7 +44,7 @@ class BaseObject(object):
 
 """)
 
-    ## generate enums
+    # generate enums
     for enum in enums:
         name = enum.__name__
         f.write("class %s:\n" % name)
@@ -59,10 +60,10 @@ class BaseObject(object):
     for klass in classes:
         name = klass.__name__
         base = "Exception" if issubclass(klass, ttypes.TExceptionBase) else "BaseObject"
-        f.write("class %s(%s):\n" % (name,  base))
+        f.write("class %s(%s):\n" % (name, base))
         f.write("\t__slots__ = %s\n\n" % klass.__slots__)
 
-        #create init
+        # create init
         args = ["self"] + ["%s=None" % x for x in klass.__slots__]
 
         f.write("\tdef __init__(%s):\n" % ", ".join(args))