def notify(topic, subject, message): configs = OmeletConfigParser() aws_key = Credential() omelet_notifier = Notifier(configs.get_sns_region(), aws_key.get_aws_access_key(), aws_key.get_aws_secret_key()) omelet_notifier.send(configs.get_sns_topic('patch'), subject, message)
def determine_region(logger): configs = OmeletConfigParser() regions = configs.get_packer_regions() logger.debug("we have %d regions : %s" % (len(regions), regions)) #config error, give up if regions < 1: logger.error("please check config.ini region setting, give up :-(") raise logger.info("use %s region" % regions[0]) return regions[0]
def get_commit_diff_info(self, repo=None): configs = OmeletConfigParser() git_repo = git.Repo(self.repo) if not repo else repo diff = list() committer = "" commits_list = list(git_repo.iter_commits(configs.get_git_branch(), max_count=10)) if len(commits_list) > 1: for commit in commits_list[0].diff(commits_list[1]): # add files if commit.a_blob and commit.a_blob.path not in diff: diff.append(commit.a_blob.path) committer = commits_list[0].committer # remove files #if commit.b_blob and commit.b_blob.path not in diff: # diff.append(commit.b_blob.path) else: raise ValueError("repo commit count is less than one") return (committer, diff)
def main(logger): #objects configs = OmeletConfigParser() region = configs.get_glacier_region() aws_key = Credential() #parse argument parser = argparse.ArgumentParser() parser.add_argument('filename', type=str, help='file to upload to Glacier') parser.add_argument('--vault', required=True, help='vault name') args = parser.parse_args() #parameters filename = args.filename vault = args.vault #exec glacier_helper = Glacier(_aws_access_key_id=aws_key.get_aws_access_key(), _aws_secret_access_key=aws_key.get_aws_secret_key(), _region=region) glacier_helper.upload(vault, filename, True)
# update my descent if any if valid: for child in db.get_tree_children(ami_name): self._launch_amipacker_series(db, child, self._get_template(child), sha1, committer, committer_mail, comment, notify, causedby) if __name__ == '__main__': parser = argparse.ArgumentParser(description='AMIPacker wrapper args') # should be ~/clone or ~/src/github.com/TrendMicroDCS parser.add_argument("-r", "--repo", type=str, help="target packer template repo master", required=True) # configs args = parser.parse_args() configs = OmeletConfigParser() # logging logging.config.fileConfig(os.path.join(configs.get_omelet_config_dir(), 'logging.ini')) logger = logging.getLogger(__name__) # aws key aws_key = Credential() # mail sender omelet_notifier = Notifier(configs.get_sns_region(), aws_key.get_aws_access_key(), aws_key.get_aws_secret_key()) try: amipacker_launcher = OmeletAMIPacker(configs, aws_key, args.repo) ret = amipacker_launcher.launch()
# # Copyright (c) 2014 TrendMicro. Data Center Services Research and Development. ([email protected]) # import os import logging import argparse import sys from distutils.sysconfig import get_python_lib sys.path.append(os.path.abspath(".")) from omelet.aws.ami import AMI from omelet.aws.credential import Credential from omelet.utils.configparser import OmeletConfigParser def main(region, id): aws_key = Credential() image = AMI(region, aws_key.get_aws_access_key(), aws_key.get_aws_secret_key()) image.get_ami_obj(id) if __name__ == '__main__': configs = OmeletConfigParser() logging.config.fileConfig(os.path.join(configs.get_omelet_config_dir(), 'logging.ini')) parser = argparse.ArgumentParser(description='Check AMI Name Args') parser.add_argument("-r", "--region", type=str, help="target region", required=True) parser.add_argument("-i", "--id", type=str, help="ami id", required=True) args = parser.parse_args() main(args.region, args.id)