From e4f2e722b2740d16ddc2ffa9b184287a7850b203 Mon Sep 17 00:00:00 2001
From: V3n3RiX <venerix@redcorelinux.org>
Date: Sun, 31 Jan 2021 01:19:56 +0000
Subject: backend : reuse binhost module in csvfiles module, one less portage
 call

---
 src/backend/check.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

(limited to 'src/backend/check.py')

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)
-- 
cgit v1.2.3