lbry-sdk/lbrynet/lbrynet_console/Settings.py

115 lines
3.3 KiB
Python
Raw Normal View History

2015-08-20 17:27:15 +02:00
import binascii
import functools
2015-08-20 17:27:15 +02:00
import json
import logging
import os
2015-08-20 17:27:15 +02:00
from twisted.internet import threads, defer
import unqlite
2015-08-20 17:27:15 +02:00
log = logging.getLogger(__name__)
def run_in_thread(fn):
@functools.wraps(fn)
def wrapped(*args, **kwargs):
return threads.deferToThread(fn, *args, **kwargs)
return wrapped
2015-08-20 17:27:15 +02:00
class LBRYSettings(object):
NAME = "settings.db"
2015-08-20 17:27:15 +02:00
def __init__(self, db_dir):
self.db_dir = db_dir
self.db = None
def start(self):
return self._open_db()
2015-08-20 17:27:15 +02:00
def stop(self):
self.db.close()
2015-08-20 17:27:15 +02:00
self.db = None
return defer.succeed(True)
def _open_db(self):
filename = os.path.join(self.db_dir, self.NAME)
log.debug("Opening %s as the settings database", filename)
self.db = unqlite.UnQLite(filename)
return defer.succeed(True)
2015-08-20 17:27:15 +02:00
@run_in_thread
2015-08-20 17:27:15 +02:00
def save_lbryid(self, lbryid):
self.db['lbryid'] = binascii.hexlify(lbryid)
self.db.commit()
2015-08-20 17:27:15 +02:00
@run_in_thread
2015-08-20 17:27:15 +02:00
def get_lbryid(self):
if 'lbryid' in self.db:
return binascii.unhexlify(self.db['lbryid'])
else:
return None
2015-08-20 17:27:15 +02:00
@run_in_thread
2015-08-20 17:27:15 +02:00
def get_server_running_status(self):
if 'server_running' in self.db:
return json.loads(self.db['server_running'])
else:
return True
2015-08-20 17:27:15 +02:00
@run_in_thread
2015-08-20 17:27:15 +02:00
def save_server_running_status(self, running):
self.db['server_running'] = json.dumps(running)
self.db.commit()
2015-08-20 17:27:15 +02:00
def get_default_data_payment_rate(self):
return self._get_payment_rate("default_data_payment_rate")
def save_default_data_payment_rate(self, rate):
return self._save_payment_rate("default_data_payment_rate", rate)
def get_server_data_payment_rate(self):
return self._get_payment_rate("server_data_payment_rate")
def save_server_data_payment_rate(self, rate):
return self._save_payment_rate("server_data_payment_rate", rate)
def get_server_crypt_info_payment_rate(self):
return self._get_payment_rate("server_crypt_info_payment_rate")
def save_server_crypt_info_payment_rate(self, rate):
return self._save_payment_rate("server_crypt_info_payment_rate", rate)
@run_in_thread
2015-08-20 17:27:15 +02:00
def _get_payment_rate(self, rate_type):
if rate_type in self.db:
return json.loads(self.db[rate_type])
else:
return None
2015-08-20 17:27:15 +02:00
@run_in_thread
2015-08-20 17:27:15 +02:00
def _save_payment_rate(self, rate_type, rate):
if rate is not None:
self.db[rate_type] = json.dumps(rate)
elif rate_type in self.db:
del self.db[rate_type]
self.db.commit()
2015-08-20 17:27:15 +02:00
@run_in_thread
2015-08-20 17:27:15 +02:00
def get_query_handler_status(self, query_identifier):
if json.dumps(('q_h', query_identifier)) in self.db:
return json.loads(self.db[(json.dumps(('q_h', query_identifier)))])
else:
return True
2015-08-20 17:27:15 +02:00
def enable_query_handler(self, query_identifier):
return self._set_query_handler_status(query_identifier, True)
def disable_query_handler(self, query_identifier):
return self._set_query_handler_status(query_identifier, False)
@run_in_thread
2015-08-20 17:27:15 +02:00
def _set_query_handler_status(self, query_identifier, status):
self.db[json.dumps(('q_h', query_identifier))] = json.dumps(status)
self.db.commit()