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