Simplify with .writable() method.

Addresses #50 (comment)
jensengroup · Jun 4, 2020 · 7085271 · 7085271
1 parent 1d17cb0
commit 7085271
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/propka/output.py b/propka/output.py
@@ -13,8 +13,7 @@ def open_file_for_writing(input_file):
         then will attempt to get file mode.
     """
     try:
-        mode = input_file.mode
-        if not ("w" in mode or "a" in mode or "+" in mode):
+        if not input_file.writable():
             raise IOError("File/stream not open for writing")
         return input_file
     except AttributeError: