From dc65254e0a249b1aadeed6570ca1937477e4e8c4 Mon Sep 17 00:00:00 2001
From: Kevin Froman <beardog@firemail.cc>
Date: Sun, 28 Jan 2018 20:02:16 -0600
Subject: [PATCH] now update block datasaved stat when saved

---
 onionr/communicator.py | 5 ++++-
 onionr/core.py         | 7 +++++++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/onionr/communicator.py b/onionr/communicator.py
index 16a44b80..9bba7c6e 100755
--- a/onionr/communicator.py
+++ b/onionr/communicator.py
@@ -137,7 +137,10 @@ class OnionrCommunicate:
             if data == False or len(data) > 10000000:
                 continue
             hasher.update(data.encode())
-            if hasher.hexdigest() == hash.strip():
+            digest = hasher.hexdigest()
+            if digest is bytes:
+                digest = digest.decode()
+            if digest == hash.strip():
                 self._core.setData(data)
                 logger.info('Successfully obtained data for ' + hash)
             else:
diff --git a/onionr/core.py b/onionr/core.py
index 2c2ef72e..96c8a7f7 100644
--- a/onionr/core.py
+++ b/onionr/core.py
@@ -169,6 +169,13 @@ class Core:
             blockFile = open(blockFileName, 'w')
             blockFile.write(data.decode())
             blockFile.close()
+
+            conn = sqlite3.connect(self.blockDB)
+            c = conn.cursor()
+            c.execute("UPDATE hashes set dataSaved=1 where id = '" + dataHash + "';")
+            conn.commit()
+            conn.close()
+
         return dataHash
 
     def dataDirEncrypt(self, password):