def __init__(self): self.file_crypt = FileCrypt() self.config = self.load_config() self.pyi_updater = PyiUpdater(self.config) self.key_handler = KeyHandler() self.package_handler = PackageHandler() self.uploader = Uploader() self.update_helpers(self.pyi_updater) helpers = { u'key_handler': self.key_handler, u'package_handler': self.package_handler, u'uploader': self.uploader, u'file_crypt': self.file_crypt, u'config': self.config, u'save': self.save_config, } self.keys_menu = keys.Keys(helpers) self.settings_menu = settings.Settings(helpers) self.sign_menu = sign.Sign(helpers) self.upload_menu = upload.Upload(helpers) header = u'Main Menu' options = [(u'Sign Updates', self.sign_menu), (u'Upload', self.upload_menu), (u'Keys', self.keys_menu), (u'Settings', self.settings_menu), (u'Quit', self.quit)] super(Worker, self).__init__(header, options)
def setup_func(): global test_data_dir config = TConfig() updater = PyiUpdater(config) ph = PackageHandler(updater) kh = KeyHandler(updater) kh.test = True ph.setup() kh.make_keys()
def make_keys(): pyi = PyiUpdater(DefaultConfig()) key_handler = KeyHandler(pyi) # Making a new set of keys # Keys will be place in the keys # Directory, in the pyi-data folder # *** Should not be ran again after # you deploy your app!!!! *** # If you need to make new keys pass # overwrite=True to make_keys key_handler.make_keys()
def test_setup(): global test_data_dir config = TConfig() updater = PyiUpdater(config) ph = PackageHandler(updater) key_dir = os.path.join(ph.data_dir, u'keys') kh = KeyHandler(updater) kh.test = True assert os.path.exists(os.path.abspath(key_dir)) assert kh.private_key_name == u'jms.pem' assert kh.public_key_name == u'jms.pub'
def setup(): # Setting up Config object default_config = DefaultConfig() # Initilizing Main object and configuring # in one step pyi = PyiUpdater(default_config) # Can also update config later pyi.update_config(default_config) # Initializing Package Handler & Key Handler # with config info package_handler = PackageHandler(pyi) key_handler = KeyHandler(pyi) # Can also be Initilized without config package_handler = PackageHandler() key_handler = KeyHandler() # Then update handlers with config later package_handler.init_app(pyi) key_handler.init_app(pyi) # Setting up work directories # Only need to run once on a new project but it's # ok if ran multipule times package_handler.setup() # Now place new packages in the folder named # "new" in the pyi-data directory # Package Archive filename should be in the form # AppName-platform-version.zip raw_input('Place updates in new folder then press enter.') # This updates the version file with the # new packages & moves them to the deploy folder. package_handler.process_packages() # This signs the update manifest & copies it # to the deploy folder key_handler.sign_update()
def setup_func2(): global pub_key global test_data_dir config = TConfig() updater = PyiUpdater(config) ph = PackageHandler(updater) kh = KeyHandler(updater) ph.setup() kh.test = True kh.make_keys() pub_key = kh.get_public_key() # Make zipfile with ChDir(test_data_dir): os.mkdir(u'test-app') with ChDir(u'test-app'): with open(u'app.txt', u'w') as f: f.write(u'I am so happy' * 1000) shutil.make_archive(u'Test App-mac-0.2.0', u'zip', u'test-app') shutil.move(u'Test App-mac-0.2.0.zip', u'new') ph.process_packages() kh.sign_update()
class Worker(Menu, CommonLogic): def __init__(self): self.file_crypt = FileCrypt() self.config = self.load_config() self.pyi_updater = PyiUpdater(self.config) self.key_handler = KeyHandler() self.package_handler = PackageHandler() self.uploader = Uploader() self.update_helpers(self.pyi_updater) helpers = { u'key_handler': self.key_handler, u'package_handler': self.package_handler, u'uploader': self.uploader, u'file_crypt': self.file_crypt, u'config': self.config, u'save': self.save_config, } self.keys_menu = keys.Keys(helpers) self.settings_menu = settings.Settings(helpers) self.sign_menu = sign.Sign(helpers) self.upload_menu = upload.Upload(helpers) header = u'Main Menu' options = [(u'Sign Updates', self.sign_menu), (u'Upload', self.upload_menu), (u'Keys', self.keys_menu), (u'Settings', self.settings_menu), (u'Quit', self.quit)] super(Worker, self).__init__(header, options) # self.menu = Menu(header, options) def update_helpers(self, pyi_updater): self.file_crypt.init_app(pyi_updater) self.key_handler.init_app(pyi_updater) self.key_handler._add_filecrypt(self.file_crypt) self.package_handler.init_app(pyi_updater) self.uploader.init_app(pyi_updater) log.debug(u'Updated helpers') def start(self): while 1: dec_path = os.path.join(cwd, u'config.data') enc_path = os.path.join(cwd, u'config.data.enc') if not os.path.exists(dec_path) and not os.path.exists(enc_path): self.initial_setup() x = self.display() if len(x) == 2: next_ = x[1] next_() # We are dynamically generating upload options # from installed plugins. So the option selected # by the user needs a little extra processing else: name = x[0] func = x[1] func(name) def quit(self): lex_file = os.path.join(cwd, u'lextab.py') yac_file = os.path.join(cwd, u'yacctab.py') if os.path.exists(lex_file): os.remove(lex_file) if os.path.exists(yac_file): os.remove(yac_file) log.debug('Quitting') print('See Ya!') sys.exit() def initial_setup(self): log.debug(u'Starting initial setup') self.display_menu_header(u'Setup Assistant') self.display_msg(u'Let\'s begin...') self.config.APP_NAME = get_correct_answer(u'Please enter app name', required=True) self.config.COMPANY_NAME = get_correct_answer(u'Please enter your ' 'company or name', required=True) self.config.DEV_DATA_DIR = cwd url = get_correct_answer(u'Enter a url to ping for updates.', required=True) self.config.UPDATE_URLS = [url] while 1: answer = ask_yes_no(u'Would you like to add another ' 'url for backup?', default='no') if answer is True: url = get_correct_answer(u'Enter another url.', required=True) self.config.UPDATE_URLS.append(url) else: break self.config.UPDATE_PATCHES = ask_yes_no(u'Would you like to enable ' 'patch updates?', default='yes') answer1 = ask_yes_no(u'Would you like to add scp settings?', default='no') answer2 = ask_yes_no(u'Would you like to add S3 settings?', default='no') if answer1: self.config.REMOTE_DIR = get_correct_answer(u'Enter remote dir', required=True) self.config.HOST = get_correct_answer(u'Enter host', required=True) self.config.USERNAME = get_correct_answer(u'Enter usernmae', required=True) key_path = get_correct_answer(u'Enter path to ssh key', required=True) # Path to private key self.config.PASSWORD = _directory_fixer(key_path) if answer2: self.config.USERNAME = get_correct_answer(u'Enter access key ID', required=True) self.config.PASSWORD = get_correct_answer(u'Enter secret ' 'Access Key', required=True) self.config.REMOTE_DIR = get_correct_answer(u'Enter bucket name', required=True) password = verify_password(u'Enter password') self.file_crypt._update_timer() self.save_config(self.config, password) self.package_handler.setup() print(u'Making keys...') self.keys_menu.make_keys() self.display_menu_header(u'Setup Complete') self.display_msg(u'Now let\'s update some apps') time.sleep(3) def save_config(self, obj, password=None): self.pyi_updater.update_config(obj) self.update_helpers(self.pyi_updater) log.debug(u'Saving Config') filename = os.path.join(cwd, u'config.data') self.file_crypt.new_file(filename) # We do this here to keep from asking users # password again when we encrypt the file if password is not None: password = self.file_crypt._gen_password(password) self.file_crypt.password = password with open(filename, 'w') as f: f.write(str(pickle.dumps(obj))) self.file_crypt.encrypt() self.write_config_py(obj) def load_config(self): log.debug(u'Loading Config') filename = os.path.join(cwd, u'config.data') salt_file = os.path.join(cwd, u'pyi-data', u'keys', u'salt') self.file_crypt.salt_file = salt_file self.file_crypt.new_file(filename) try: self.file_crypt.decrypt() except FileCryptPasswordError: sys.exit(u'Failed password attempt') except Exception as e: log.error(str(e), exc_info=True) log.warning(u'No enc file. Will try to load plain config') try: with open(filename, 'r') as f: config_data = pickle.loads(f.read()) self.file_crypt.encrypt() except Exception as e: log.error(e, exc_info=True) config_data = SetupConfig() return config_data def write_config_py(self, obj): filename = os.path.join(cwd, u'client_config.py') attr_str_format = " {} = '{}'\n" attr_format = " {} = {}\n" with open(filename, u'w') as f: f.write('class ClientConfig(object):\n') if hasattr(obj, 'APP_NAME') and obj.APP_NAME is not None: f.write(attr_str_format.format('APP_NAME', obj.APP_NAME)) if hasattr(obj, 'COMPANY_NAME') and obj.COMPANY_NAME is not None: f.write(attr_str_format.format('COMPANY_NAME', obj.COMPANY_NAME)) if hasattr(obj, 'UPDATE_URLS') and obj.UPDATE_URLS is not None: f.write(attr_format.format('UPDATE_URLS', obj.UPDATE_URLS)) if hasattr(obj, 'PUBLIC_KEY') and obj.PUBLIC_KEY is not None: f.write(attr_str_format.format('PUBLIC_KEY', obj.PUBLIC_KEY))
class Worker(Menu, CommonLogic): def __init__(self): self.file_crypt = FileCrypt() self.config = self.load_config() self.pyi_updater = PyiUpdater(self.config) self.key_handler = KeyHandler() self.package_handler = PackageHandler() self.uploader = Uploader() self.update_helpers(self.pyi_updater) helpers = { u'key_handler': self.key_handler, u'package_handler': self.package_handler, u'uploader': self.uploader, u'file_crypt': self.file_crypt, u'config': self.config, u'save': self.save_config, } self.keys_menu = keys.Keys(helpers) self.settings_menu = settings.Settings(helpers) self.sign_menu = sign.Sign(helpers) self.upload_menu = upload.Upload(helpers) header = u'Main Menu' options = [(u'Sign Updates', self.sign_menu), (u'Upload', self.upload_menu), (u'Keys', self.keys_menu), (u'Settings', self.settings_menu), (u'Quit', self.quit)] super(Worker, self).__init__(header, options) # self.menu = Menu(header, options) def update_helpers(self, pyi_updater): self.file_crypt.init_app(pyi_updater) self.key_handler.init_app(pyi_updater) self.key_handler._add_filecrypt(self.file_crypt) self.package_handler.init_app(pyi_updater) self.uploader.init_app(pyi_updater) log.debug(u'Updated helpers') def start(self): while 1: dec_path = os.path.join(cwd, u'config.data') enc_path = os.path.join(cwd, u'config.data.enc') if not os.path.exists(dec_path) and not os.path.exists(enc_path): self.initial_setup() x = self.display() if len(x) == 2: next_ = x[1] next_() # We are dynamically generating upload options # from installed plugins. So the option selected # by the user needs a little extra processing else: name = x[0] func = x[1] func(name) def quit(self): lex_file = os.path.join(cwd, u'lextab.py') yac_file = os.path.join(cwd, u'yacctab.py') if os.path.exists(lex_file): os.remove(lex_file) if os.path.exists(yac_file): os.remove(yac_file) log.debug('Quitting') print('See Ya!') sys.exit() def initial_setup(self): log.debug(u'Starting initial setup') self.display_menu_header(u'Setup Assistant') self.display_msg(u'Let\'s begin...') self.config.APP_NAME = get_correct_answer(u'Please enter app name', required=True) self.config.COMPANY_NAME = get_correct_answer( u'Please enter your ' 'company or name', required=True) self.config.DEV_DATA_DIR = cwd url = get_correct_answer(u'Enter a url to ping for updates.', required=True) self.config.UPDATE_URLS = [url] while 1: answer = ask_yes_no( u'Would you like to add another ' 'url for backup?', default='no') if answer is True: url = get_correct_answer(u'Enter another url.', required=True) self.config.UPDATE_URLS.append(url) else: break self.config.UPDATE_PATCHES = ask_yes_no( u'Would you like to enable ' 'patch updates?', default='yes') answer1 = ask_yes_no(u'Would you like to add scp settings?', default='no') answer2 = ask_yes_no(u'Would you like to add S3 settings?', default='no') if answer1: self.config.REMOTE_DIR = get_correct_answer(u'Enter remote dir', required=True) self.config.HOST = get_correct_answer(u'Enter host', required=True) self.config.USERNAME = get_correct_answer(u'Enter usernmae', required=True) key_path = get_correct_answer(u'Enter path to ssh key', required=True) # Path to private key self.config.PASSWORD = _directory_fixer(key_path) if answer2: self.config.USERNAME = get_correct_answer(u'Enter access key ID', required=True) self.config.PASSWORD = get_correct_answer( u'Enter secret ' 'Access Key', required=True) self.config.REMOTE_DIR = get_correct_answer(u'Enter bucket name', required=True) password = verify_password(u'Enter password') self.file_crypt._update_timer() self.save_config(self.config, password) self.package_handler.setup() print(u'Making keys...') self.keys_menu.make_keys() self.display_menu_header(u'Setup Complete') self.display_msg(u'Now let\'s update some apps') time.sleep(3) def save_config(self, obj, password=None): self.pyi_updater.update_config(obj) self.update_helpers(self.pyi_updater) log.debug(u'Saving Config') filename = os.path.join(cwd, u'config.data') self.file_crypt.new_file(filename) # We do this here to keep from asking users # password again when we encrypt the file if password is not None: password = self.file_crypt._gen_password(password) self.file_crypt.password = password with open(filename, 'w') as f: f.write(str(pickle.dumps(obj))) self.file_crypt.encrypt() self.write_config_py(obj) def load_config(self): log.debug(u'Loading Config') filename = os.path.join(cwd, u'config.data') salt_file = os.path.join(cwd, u'pyi-data', u'keys', u'salt') self.file_crypt.salt_file = salt_file self.file_crypt.new_file(filename) try: self.file_crypt.decrypt() except FileCryptPasswordError: sys.exit(u'Failed password attempt') except Exception as e: log.error(str(e), exc_info=True) log.warning(u'No enc file. Will try to load plain config') try: with open(filename, 'r') as f: config_data = pickle.loads(f.read()) self.file_crypt.encrypt() except Exception as e: log.error(e, exc_info=True) config_data = SetupConfig() return config_data def write_config_py(self, obj): filename = os.path.join(cwd, u'client_config.py') attr_str_format = " {} = '{}'\n" attr_format = " {} = {}\n" with open(filename, u'w') as f: f.write('class ClientConfig(object):\n') if hasattr(obj, 'APP_NAME') and obj.APP_NAME is not None: f.write(attr_str_format.format('APP_NAME', obj.APP_NAME)) if hasattr(obj, 'COMPANY_NAME') and obj.COMPANY_NAME is not None: f.write( attr_str_format.format('COMPANY_NAME', obj.COMPANY_NAME)) if hasattr(obj, 'UPDATE_URLS') and obj.UPDATE_URLS is not None: f.write(attr_format.format('UPDATE_URLS', obj.UPDATE_URLS)) if hasattr(obj, 'PUBLIC_KEY') and obj.PUBLIC_KEY is not None: f.write(attr_str_format.format('PUBLIC_KEY', obj.PUBLIC_KEY))
import shutil import sys from jms_utils.paths import ChDir sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) from pyi_updater import PyiUpdater from pyi_updater.key_handler import KeyHandler from pyi_updater.package_handler import PackageHandler from pyi_updater.utils import remove_dot_files from tconfig import TConfig pyi_updater = PyiUpdater(TConfig()) kh = KeyHandler(pyi_updater) kh.test = True kh._add_filecrypt() ph = PackageHandler(pyi_updater) PYI_DATA = os.path.abspath(os.path.join(u'tests', u'pyi-data')) def setup_func(): ph.setup() kh.make_keys() test_data_dir = os.path.abspath(os.path.join(u'tests', u'test data', u'5.0')) with ChDir(test_data_dir): files = remove_dot_files(os.listdir(os.getcwd())) for f in files:
import shutil import sys from jms_utils.paths import ChDir sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) from pyi_updater import PyiUpdater from pyi_updater.key_handler import KeyHandler from pyi_updater.package_handler import PackageHandler from pyi_updater.utils import remove_dot_files from tconfig import TConfig pyi_updater = PyiUpdater(TConfig()) kh = KeyHandler(pyi_updater) kh.test = True kh._add_filecrypt() ph = PackageHandler(pyi_updater) PYI_DATA = os.path.abspath(os.path.join(u'tests', u'pyi-data')) def setup_func(): ph.setup() kh.make_keys() test_data_dir = os.path.abspath( os.path.join(u'tests', u'test data', u'5.0')) with ChDir(test_data_dir): files = remove_dot_files(os.listdir(os.getcwd())) for f in files: