Add read/write file helpers to Generator

funkybob · Mar 6, 2017 · 6f9496e · 6f9496e
1 parent 0cc51b6
commit 6f9496e
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/graaf/base.py b/graaf/base.py
@@ -1,3 +1,4 @@
+import io
 import os
 
 import yaml
@@ -19,6 +20,14 @@ def process(self, src_dir, dest_dir, filename, processor):
         '''
         raise NotImplementedError
 
+    def read_file(self, src_dir, filename):
+        with io.open(os.path.join(src_dir, filename), mode='r', encoding='utf-8') as fin:
+            return fin.read()
+
+    def write_file(self, dest_dir, filename, content):
+        with io.open(os.path.join(dest_dir, filename), mode='w', encoding='utf-8') as fout:
+            fout.write(content)
+
 
 def get_yaml(src):
     '''