def check_database(self):
self.bitHopper.log_msg('DB: Checking Database')
if os.path.exists(DB_FN):
try:
versionfd = open(VERSION_FN, 'rb')
version = versionfd.read()
self.bitHopper.log_dbg("DB: Version " + version)
if version == "0.1":
self.bitHopper.log_msg('DB: Old Database')
versionfd.close()
except:
os.remove(DB_FN)
version = open(VERSION_FN, 'wb')
version.write("0.2")
version.close()
self.database = sqlite3.connect(DB_FN)
self.curs = self.database.cursor()
if version == "0.1":
sql = "SELECT name FROM sqlite_master WHERE type='table'"
self.curs.execute(sql)
result = self.curs.fetchall()
for item in result:
sql = "ALTER TABLE " + item[0] + " ADD COLUMN user TEXT"
self.curs.execute(sql)
self.database.commit()
def check_database(self):
logging.info('Checking Database')
if os.path.exists(DB_FN):
try:
versionfd = open(VERSION_FN, 'rb')
version = versionfd.read()
logging.debug("DB: Version " + version)
if version == "0.1":
logging.info('Old Database')
versionfd.close()
except:
os.remove(DB_FN)
version = open(VERSION_FN, 'wb')
version.write("0.2")
version.close()
self.database = sqlite3.connect(DB_FN)
self.curs = self.database.cursor()
if version == "0.1":
sql = "SELECT name FROM sqlite_master WHERE type='table'"
self.curs.execute(sql)
result = self.curs.fetchall()
for item in result:
sql = "ALTER TABLE " + item[0] + " ADD COLUMN user TEXT"
self.curs.execute(sql)
self.database.commit()
def check_database(self):
self.bitHopper.log_msg('Checking Database')
if os.path.exists(DB_FN):
try:
versionfd = open(VERSION_FN, 'rb')
version = versionfd.read()
self.bitHopper.log_dbg("DB Verson: " + version)
if version == "0.1":
self.bitHopper.log_msg('Old Database')
versionfd.close()
except:
os.remove(DB_FN)
version = open(VERSION_FN, 'wb')
version.write("0.2")
version.close()
self.database = sqlite3.connect(DB_FN)
self.curs = self.database.cursor()
if version == "0.1":
sql = "SELECT name FROM sqlite_master WHERE type='table'"
self.curs.execute(sql)
result = self.curs.fetchall()
for item in result:
sql = "ALTER TABLE " + item[0] + " ADD COLUMN user TEXT"
self.curs.execute(sql)
self.database.commit()
for server_name in self.pool.get_servers():
sql = "CREATE TABLE IF NOT EXISTS "+server_name +" (diff REAL, shares INTEGER, rejects INTEGER, stored_payout REAL, user TEXT)"
self.curs.execute(sql)
self.database.commit()
def gen_config():
for item in os.listdir(env.get('dir_config')):
_path = runtime.path.join(env.get('dir_config'), item)
if os.path.isfile(_path):
os.remove(_path)
else:
shutil.rmtree(_path)
for bundle_desc in runtime.path.all_bundles_desc():
ENV_DICT = env.all_dict()
with open(bundle_desc, 'r') as fp:
_bundle_desc = yaml.load(fp.read())
_bundle = store.find(Bundle, Bundle.name == _bundle_desc['name']).one()
# _bundle = bundle.get(_bundle_desc['name'])
ENV_DICT.update(_bundle.env())
for _conf in _bundle_desc.get('conf', []):
src = _conf['src'].format(**ENV_DICT)
dst = _conf['dst'].format(**ENV_DICT)
dst_dir = os.path.dirname(dst)
if not os.path.exists(dst_dir):
os.makedirs(dst_dir)
content = None
with open(src, 'r') as fp:
try:
content = template.Template(fp.read()).generate(**ENV_DICT)
except Exception, e:
print e
print traceback.format_exc()
with open(dst, 'w') as fp:
fp.write(content)
# print dst
logger.info('genrate config file %s' % dst)
def serve(self):
try:
os.remove(self.server_address)
except OSError:
pass
server = listen(self.server_address, family=socket.AF_UNIX)
while True:
try:
new_sock, address = server.accept()
self.pool.spawn_n(self.handle, new_sock.makefile('rw'))
except (SystemExit, KeyboardInterrupt):
break
