backend : reuse binhost module in csvfiles module, one less portage call

1 files changed, 23 insertions, 0 deletions

diff --git a/src/backend/check.py b/src/backend/check.py
index b2d8206..d5d3aca 100644
--- a/src/backend/check.py
+++ b/src/backend/check.py
@@ -2,11 +2,34 @@
 
 import os
 import subprocess
+import sisyphus.binhost
 import sisyphus.filesystem
 
 def root():
     return True if os.getuid() == 0 else False
 
+
+def branch():
+    binhostURL = sisyphus.binhost.getURL()
+    localBranch = subprocess.check_output(['git', 'rev-parse', '--abbrev-ref', 'HEAD'])
+    mismatch = int()
+
+    if "packages-next" in binhostURL:
+        os.chdir(sisyphus.filesystem.portageRepoDir)
+        if localBranch.decode().strip() == "next":
+            print(binhostURL.replace('packages-next', 'csv-next') + 'remotePackagesPre.csv')
+            mismatch = int(0)
+        else:
+            mismatch = int(1)
+    else:
+        if localBranch.decode().strip() == "master":
+            mismatch = int(0)
+        else:
+            mismatch = int(1)
+
+    return localBranch,mismatch
+branch()
+
 def portage():
     if os.path.isdir(os.path.join(sisyphus.filesystem.portageRepoDir, '.git')):
         os.chdir(sisyphus.filesystem.portageRepoDir)