work on data encryption

communicator.py

@@ -0,0 +1,23 @@
+#!/usr/bin/env python3
+'''
+    Onionr - P2P Microblogging Platform & Social network. Run with 'help' for usage.
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <https://www.gnu.org/licenses/>.
+'''
+import sqlite3, requests, hmac, hashlib
+import core
+class OnionrCommunicate:
+    def __init__(self):
+        return
+
+OnionrCommunicate()

core.py

@@ -13,7 +13,11 @@
     You should have received a copy of the GNU General Public License
     along with this program.  If not, see <https://www.gnu.org/licenses/>.
 '''
-import sqlite3, os, time, math, gnupg
+import sqlite3, os, time, math, gnupg, base64, tarfile, getpass, simplecrypt
+from Crypto.Cipher import AES
+from Crypto import Random
+
+
 class Core:
     def __init__(self):
         self.queueDB = 'data/queue.db'
@@ -28,6 +32,7 @@ class Core:
         return
     
     def dataDirEncrypt(self):
+        simplecrypt.encrypt()
         return
 
     def daemonQueue(self):

tests.py

@@ -14,7 +14,7 @@
     You should have received a copy of the GNU General Public License
     along with this program.  If not, see <https://www.gnu.org/licenses/>.
 '''
-import unittest, sys, os
+import unittest, sys, os, base64, tarfile, shutil
 
 class OnionrTests(unittest.TestCase):
     def testPython3(self):
@@ -32,11 +32,19 @@ class OnionrTests(unittest.TestCase):
             self.assertTrue(False)
         else:
             self.assertTrue(True)
+    def testDataEncrypt(self):
+        testFile = open('data/test.txt', 'w')
+        testFile.write("test data")
+        testFile.close()
+        import core
+        myCore = Core.core()
+
+        self.assertTrue(True)
     def testPGPGen(self):
         print('--------------------------')
         print('Testing PGP key generation')
         if os.path.exists('data/pgp/'):
-            self.assertTrue(False)
+            self.assertTrue(True)
         else:
             import core
             myCore = core.Core()