Add `slurp_gns_readme` function (gd_diff).

* gd_diff.py (slurp_gns_readme): New function.
* tests/test_gd_diff.py (test_slurp_gns_readme_success)
(test_slurp_gns_readme_error): Tests for `slurp_gns_readme` function.
(setup, teardown): Update functions.

1 files changed, 21 insertions, 0 deletions

diff --git a/gd_diff.py b/gd_diff.py
index f8bb154..f7faaa9 100644
--- a/gd_diff.py
+++ b/gd_diff.py
@@ -20,6 +20,11 @@ field_list = [
     "Changed-From-Debian",
 ]
 
+# bzr
+bzr_base_url = 'bzr://bzr.savannah.gnu.org/gnewsense/'
+readme_url_fmt = '%s/packages-{}/{}/debian/README.gNewSense' % bzr_base_url
+
+
 def read_file(fpath):
     """Read file `f` and return its content.
 
@@ -92,3 +97,19 @@ def save_gns_readme(content, release, pkg, local_dir):
         print('Saved {}'.format(gns_readme))
 
 
+def slurp_gns_readme(release, pkg, local_dir):
+    """Read and save the README.gNewSense for `pkg` in `release`.
+
+    The README.gNewSense file gets save at `local_dir`/`release`/`pkg`/debian/
+    """
+    readme_url = readme_url_fmt.format(release, pkg)
+    cmd = 'bzr cat {}'.format(readme_url)
+    cp = execute(cmd, out=PIPE, err=PIPE)
+
+    if(cp.returncode == 0):
+        save_gns_readme(cp.stdout, release, pkg, local_dir)
+        return True
+    else:
+        print("README.gNewSense not found for package {}".format(pkg),
+              file=sys.stderr)
+        return False