From e62182eb57ade6da31e7d52c0ab8e033dcb7e627 Mon Sep 17 00:00:00 2001
From: dave <dave_gitlab@davepedu.com>
Date: Tue, 29 Dec 2015 23:25:23 -0800
Subject: [PATCH] Fix 404 check

---
 datadb/datadb.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/datadb/datadb.py b/datadb/datadb.py
index b4001b8..45214af 100755
--- a/datadb/datadb.py
+++ b/datadb/datadb.py
@@ -32,8 +32,8 @@ def restore(profile, conf, force=False): #remote_uri, local_dir, identity='/root
     original_perms = stat(conf["dir"])
     dest = urlparse(conf["uri"])
     
-    status_code = head(DATADB_HTTP_API+'get_backup', params={'proto':'rsync', 'name':profile}).status_code
-    if status_code:
+    status_code = head(DATADB_HTTP_API+'get_backup', params={'proto':dest.scheme, 'name':profile}).status_code
+    if status_code == 404:
         print("Connected to datadb, but datasource '{}' doesn't exist. Exiting".format(profile))
         # TODO: special exit code >1 to indicate this?
         return