class Starter(object):
def __init__(self):
self.manager = RecipesManager()
def check(self):
if not self.manager.list():
logger.info(" :: I think it's your first time with sushi")
logger.info(" :: Can I suggest you simple basic recipe ?")
return True
return False
def run(self):
url = 'https://github.com/Socketubs/Sushi/raw/master/recipes/basic.tar.gz'
logger.info(' :: Searching')
r = requests.get(url, prefetch=False)
try:
r.raise_for_status()
except Exception as err:
raise StarterException(
"Sorry I can't find this old recipe in my cookbook. Try later. (%s)"
% err)
src = os.path.join(conf.get('paths', 'sushi_recipes'), 'basic.tar.gz')
dst = conf.get('paths', 'sushi_recipes')
with open(src, 'w') as f:
for buf in r.iter_content(1024):
if buf:
f.write(buf)
logger.info(' :: Teaching')
# Open tarfile
tar = tarfile.open(src, 'r:gz')
if tarfile.is_tarfile(src):
tar.extractall(dst)
else:
raise Exception('Archive invalid (not a gzipped tarfile)')
# Remove archive
os.remove(src)
logger.info(' :: Done')
class Starter(object):
def __init__(self):
self.manager = RecipesManager()
def check(self):
if not self.manager.list():
logger.info(" :: I think it's your first time with sushi")
logger.info(" :: Can I suggest you simple basic recipe ?")
return True
return False
def run(self):
url = 'https://github.com/Socketubs/Sushi/raw/master/recipes/basic.tar.gz'
logger.info(' :: Searching')
r = requests.get(url, prefetch=False)
try:
r.raise_for_status()
except Exception as err:
raise StarterException("Sorry I can't find this old recipe in my cookbook. Try later. (%s)" % err)
src = os.path.join(conf.get('paths', 'sushi_recipes'), 'basic.tar.gz')
dst = conf.get('paths', 'sushi_recipes')
with open(src, 'w') as f:
for buf in r.iter_content(1024):
if buf:
f.write(buf)
logger.info(' :: Teaching')
# Open tarfile
tar = tarfile.open(src, 'r:gz')
if tarfile.is_tarfile(src):
tar.extractall(dst)
else:
raise Exception('Archive invalid (not a gzipped tarfile)')
# Remove archive
os.remove(src)
logger.info(' :: Done')
def start(self):
###################
# craft #
###################
if self.args.get('craft', False):
logger.info('==> Craft your project')
path = self.args.get('<path>')
recipe = self.args.get('--recipe', False)
if not recipe:
recipe = conf.get('settings', 'recipe', 'default')
logger.info(' => Recipe: %s' % recipe)
try:
unbundle(recipe, path)
except Exception as err:
logger.info('Error: %s (%s)' % (err, recipe))
sys.exit(1)
logger.info('==> Call helpers')
run_helpers(recipe, path)
logger.info('==> Done')
###################
# learn #
###################
elif self.args.get('learn', False):
recipes = self.args.get('<name>')
manager = RecipesManager()
try:
for recipe in recipes:
logger.info('==> Learn %s' % recipe)
manager.add(recipe)
except RecipeAlreadyLearn as err:
logger.error('Error: %s' % err)
sys.exit(1)
logger.info('==> Done')
###################
# forget #
###################
elif self.args.get('forget', False):
recipes = self.args.get('<name>')
manager = RecipesManager()
try:
for recipe in recipes:
logger.info('==> Forget %s' % recipe)
manager.delete(recipe)
logger.info('==> Done')
except Exception as err:
logger.error('Error: %s' % err)
sys.exit(1)
###################
# list #
###################
elif self.args.get('list', False):
manager = RecipesManager()
recipes = manager.list()
if not recipes:
logger.info('==> No recipe learned')
sys.exit(1)
logger.info('==> Recipes learned')
for recipe in recipes:
logger.info(recipe)
sys.exit(0)
###################
# all #
###################
elif self.args.get('all', False):
manager = RecipesManager()
recipes = manager.list_available()
if not recipes:
logger.info('==> No recipe available')
sys.exit(1)
logger.info('==> Recipes available')
for recipe in recipes:
logger.info(recipe)
sys.exit(0)
###################
# upgrade #
###################
elif self.args.get('upgrade', False):
cb = Cookbook()
logger.info('==> Update cookbooks')
cookbooks = cb.upgrade(ignore_errors=True)
###################
# cookbook #
###################
elif self.args.get('cookbook', False):
cb = Cookbook()
cookbooks = cb.list()
if not cookbooks:
logger.info('==> No cookbooks registered')
sys.exit(1)
logger.info('==> Cookbooks')
for cookbook in cookbooks:
logger.info(cookbook)
sys.exit(0)
###################
# cookbook-add #
###################
elif self.args.get('cookbook-add', False):
repo_name = self.args.get('<name>')[0]
logger.info('==> Add %s cookbook' % repo_name)
cb = Cookbook()
try:
cb.add(repo_name)
logger.info('==> Done')
except CookbookException as err:
logger.info('Error: %s' % err)
sys.exit(1)
sys.exit(0)
###################
# cookbook-del #
###################
elif self.args.get('cookbook-del', False):
repo_name = self.args.get('<name>')[0]
logger.info('==> Delete %s cookbook' % repo_name)
cb = Cookbook()
if confirm():
try:
cb.remove(repo_name)
logger.info('==> Done')
except CookbookException as err:
logger.info('Error: %s' % err)
sys.exit(1)
else:
print('Abort.')
sys.exit(1)
sys.exit(0)
sys.exit(100)
def start(self):
###################
# craft #
###################
if self.args.get('craft', False):
logger.info('==> Craft your project')
path = self.args.get('<path>')
recipe = self.args.get('--recipe', False)
if not recipe:
recipe = conf.get('settings', 'recipe', 'default')
logger.info(' => Recipe: %s' % recipe)
try:
unbundle(recipe, path)
except Exception as err:
logger.info('!! %s (%s)' % (err.message, recipe))
sys.exit(1)
logger.info('==> Call helpers')
run_helpers(recipe, path)
logger.info('==> Done')
###################
# learn #
###################
elif self.args.get('learn', False):
recipes = self.args.get('<name>')
manager = RecipesManager()
try:
for recipe in recipes:
logger.info('==> Learn %s' % recipe)
manager.add(recipe)
except RecipeAlreadyLearn as err:
logger.error('!! %s' % err.message)
sys.exit(1)
except RecipeUnvailable as err:
logger.error('!! %s' % err.message)
sys.exit(1)
logger.info('==> Done')
###################
# forget #
###################
elif self.args.get('forget', False):
recipes = self.args.get('<name>')
manager = RecipesManager()
try:
for recipe in recipes:
logger.info('==> Forget %s' % recipe)
manager.delete(recipe)
logger.info('==> Done')
except Exception as err:
logger.error('!! %s' % err.message)
sys.exit(1)
###################
# list #
###################
elif self.args.get('list', False):
manager = RecipesManager()
recipes = manager.list()
if not recipes:
logger.info('==> No recipe learned')
sys.exit(1)
logger.info('==> Recipes learned')
for recipe in recipes:
logger.info(recipe)
sys.exit(0)
###################
# all #
###################
elif self.args.get('all', False):
manager = RecipesManager()
recipes = manager.list_available()
if not recipes:
logger.info('==> No recipe available')
sys.exit(1)
logger.info('==> Recipes available')
for recipe in recipes:
logger.info(recipe)
sys.exit(0)
###################
# upgrade #
###################
elif self.args.get('upgrade', False):
cb = Cookbook()
logger.info('==> Update cookbooks')
cookbooks = cb.upgrade(ignore_errors=True)
###################
# cookbook #
###################
elif self.args.get('cookbook', False):
cb = Cookbook()
cookbooks = cb.list()
if not cookbooks:
logger.info('==> No cookbooks registered')
sys.exit(1)
logger.info('==> Cookbooks')
for cookbook in cookbooks:
logger.info(cookbook)
sys.exit(0)
###################
# cookbook-add #
###################
elif self.args.get('cookbook-add', False):
repo_name = self.args.get('<name>')[0]
logger.info('==> Add %s cookbook' % repo_name)
cb = Cookbook()
try:
cb.add(repo_name)
logger.info('==> Done')
except CookbookException as err:
logger.info('!! %s' % err.message)
sys.exit(1)
sys.exit(0)
###################
# cookbook-del #
###################
elif self.args.get('cookbook-del', False):
repo_name = self.args.get('<name>')[0]
logger.info('==> Delete %s cookbook' % repo_name)
cb = Cookbook()
if confirm():
try:
cb.remove(repo_name)
logger.info('==> Done')
except CookbookException as err:
logger.info('!! %s' % err.message)
sys.exit(1)
else:
print('Abort.')
sys.exit(1)
sys.exit(0)
sys.exit(100)
