work on blacklisting data

This commit is contained in:
Kevin Froman 2018-08-10 02:03:49 -05:00
parent be77e02835
commit 1e37684163
No known key found for this signature in database
GPG Key ID: 0D414D0FE405B63B
2 changed files with 75 additions and 0 deletions

62
onionr/onionrblacklist.py Normal file
View File

@ -0,0 +1,62 @@
'''
Onionr - P2P Microblogging Platform & Social network.
This file handles maintenence of a blacklist database, for blocks and peers
'''
'''
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, os
class OnionrBlackList:
def __init__(self, coreInst):
self.blacklistDB = 'data/blacklist.db'
self._core = coreInst
if not os.path.exists(self.blacklistDB):
self.generateDB()
return
def inBlacklist(self, data):
return
def _dbExecute(self, toExec):
conn = sqlite3.connect(self.blacklistDB)
c = conn.cursor()
c.execute(toExec)
conn.commit()
conn.close()
def deleteBeforeDate(self, date):
# TODO, delete blacklist entries before date
return
def generateDB(self):
self._dbExecute'''CREATE TABLE blacklist(
hash text primary key not null,
type text
);
''')
return
def clearDB(self):
self._dbExecute('''delete from blacklist;);''')
def addToDB(self, data):
hashed = self._core._crypto.sha3Hash(data)
if not hashed.isalnum():
raise Exception("Hashed data is not alpha numeric")
insert = (hashed,)
self._dbExecute('insert into blacklist (hash) VALUES(' + data + ');')

View File

@ -598,6 +598,19 @@ class OnionrUtils:
self.powSalt = retData self.powSalt = retData
return retData return retData
def strToBytes(self, data):
try:
data = data.encode()
except AttributeError:
pass
return data
def bytesToStr(self, data):
try:
data = data.decode()
except AttributeError:
pass
return data
def size(path='.'): def size(path='.'):
''' '''
Returns the size of a folder's contents in bytes Returns the size of a folder's contents in bytes