Onionr/onionr/onionrstorage/setdata.py

39 lines
1.3 KiB
Python
Raw Normal View History

2019-06-22 22:54:41 +00:00
import sys, sqlite3
2019-07-19 19:49:56 +00:00
import onionrstorage, onionrexceptions, onionrcrypto
import filepaths, storagecounter
from coredb import dbfiles
def set_data(data):
2019-06-22 22:54:41 +00:00
'''
Set the data assciated with a hash
'''
2019-07-19 19:49:56 +00:00
crypto = onionrcrypto.OnionrCrypto()
storage_counter = storagecounter.StorageCounter()
2019-06-22 22:54:41 +00:00
data = data
dataSize = sys.getsizeof(data)
if not type(data) is bytes:
data = data.encode()
2019-07-19 19:49:56 +00:00
dataHash = crypto.sha3Hash(data)
2019-06-22 22:54:41 +00:00
if type(dataHash) is bytes:
dataHash = dataHash.decode()
2019-07-19 19:49:56 +00:00
blockFileName = filepaths.block_data_location + dataHash + '.dat'
2019-06-22 22:54:41 +00:00
try:
2019-07-19 19:49:56 +00:00
onionrstorage.getData(dataHash)
2019-06-22 22:54:41 +00:00
except onionrexceptions.NoDataAvailable:
2019-07-19 19:49:56 +00:00
if storage_counter.addBytes(dataSize) != False:
onionrstorage.store(data, blockHash=dataHash)
conn = sqlite3.connect(dbfiles.block_meta_db, timeout=30)
2019-06-22 22:54:41 +00:00
c = conn.cursor()
c.execute("UPDATE hashes SET dataSaved=1 WHERE hash = ?;", (dataHash,))
conn.commit()
conn.close()
2019-07-19 19:49:56 +00:00
with open(filepaths.data_nonce_file, 'a') as nonceFile:
2019-06-22 22:54:41 +00:00
nonceFile.write(dataHash + '\n')
else:
raise onionrexceptions.DiskAllocationReached
else:
2019-06-29 18:18:31 +00:00
raise onionrexceptions.DataExists("Data is already set for " + dataHash)
2019-06-22 22:54:41 +00:00
return dataHash