protocols: get_from_file: allow passing extension

Allow passing the name of the extension if protocol object was
created with get_from_file().

Author: Tzafrir Cohen

knesset_data/protocols/base.py

@@ -15,8 +15,9 @@ class BaseProtocolFile(object):
 
     temp_file_suffix = "temp_knesset_data_protocols_"
 
-    def __init__(self, file, proxies=None):
+    def __init__(self, file, proxies=None, extension=None):
         self._file_type, self._file_data = file
+        self.extension = extension
         self._cleanup = []
         self._proxies = proxies if proxies else {}
 
@@ -55,6 +56,8 @@ def file_extension(self):
         if self._file_type in ("filename", "url") and self._file_data:
             filename, file_extension = os.path.splitext(self._file_data)
             return file_extension[1:]
+        if self.extension is not None:
+            return self.extension
         else:
             return None
 
@@ -91,8 +94,8 @@ def _close(self):
 
     @classmethod
     @contextlib.contextmanager
-    def _get_from(cls, file_type, file_data, proxies=None):
-        obj = cls((file_type, file_data), proxies=proxies)
+    def _get_from(cls, file_type, file_data, proxies=None, extension=None):
+        obj = cls((file_type, file_data), proxies=proxies, extension=extension)
         try:
             yield obj
         finally:
@@ -110,8 +113,8 @@ def get_from_url(cls, url, proxies=None):
 
     @classmethod
     @contextlib.contextmanager
-    def get_from_file(cls, file):
-        with cls._get_from('file', file) as p: yield p
+    def get_from_file(cls, file, extension=None):
+        with cls._get_from('file', file, extension=extension) as p: yield p
 
     @classmethod
     @contextlib.contextmanager