def main():
"""Application entry point.
"""
configure_logging()
plugin_manager = create_plugin_manager()
config = PiConfigParser("~/.config/pibooth/pibooth.cfg", plugin_manager)
counters = Counters(config.join_path("counters.pickle"),
taken=0,
printed=0,
forgotten=0,
remaining_duplicates=config.getint(
'PRINTER', 'max_duplicates'))
if '--json' in sys.argv:
print(json.dumps(counters.data))
elif '--update' in sys.argv:
try:
print("\nUpdating counters (current value in square bracket):\n")
for name in counters:
value = input(" -> {:.<18} [{:>4}] : ".format(
name.capitalize(), counters[name]))
if value.strip():
setattr(counters, name, int(value))
except KeyboardInterrupt:
pass
print()
else:
print("\nListing current counters:\n")
for name in counters:
print(" -> {:.<25} : {:>4}".format(name.capitalize(),
counters[name]))
print()
def main():
"""Application entry point.
"""
configure_logging()
plugin_manager = create_plugin_manager()
config = PiConfigParser("~/.config/pibooth/pibooth.cfg", plugin_manager)
# Register plugins
plugin_manager.load_all_plugins(
config.gettuple('GENERAL', 'plugins', 'path'),
config.gettuple('GENERAL', 'plugins_disabled', str))
LOGGER.info(
"Installed plugins: %s", ", ".join([
plugin_manager.get_friendly_name(p)
for p in plugin_manager.list_external_plugins()
]))
# Update configuration with plugins ones
plugin_manager.hook.pibooth_configure(cfg=config)
# Initialize varibales normally done by the app
picture_plugin = plugin_manager.get_plugin('pibooth-core:picture')
picture_plugin.texts_vars['date'] = datetime.now()
picture_plugin.texts_vars['count'] = Counters(
config.join_path("counters.pickle"),
taken=0,
printed=0,
forgotten=0,
remaining_duplicates=config.getint('PRINTER', 'max_duplicates'))
for path in config.gettuple('GENERAL', 'directory', 'path'):
regenerate_all_images(plugin_manager, config, path)
def __init__(self, config, plugin_manager):
self._pm = plugin_manager
self._config = config
# Create directories where pictures are saved
for savedir in config.gettuple('GENERAL', 'directory', 'path'):
if osp.isdir(savedir) and config.getboolean('GENERAL', 'debug'):
shutil.rmtree(savedir)
if not osp.isdir(savedir):
os.makedirs(savedir)
# Prepare the pygame module for use
os.environ['SDL_VIDEO_CENTERED'] = '1'
pygame.init()
# Create window of (width, height)
init_size = self._config.gettyped('WINDOW', 'size')
init_debug = self._config.getboolean('GENERAL', 'debug')
init_color = self._config.gettyped('WINDOW', 'background')
init_text_color = self._config.gettyped('WINDOW', 'text_color')
if not isinstance(init_color, (tuple, list)):
init_color = self._config.getpath('WINDOW', 'background')
title = 'Pibooth v{}'.format(pibooth.__version__)
if not isinstance(init_size, str):
self._window = PtbWindow(title,
init_size,
color=init_color,
text_color=init_text_color,
debug=init_debug)
else:
self._window = PtbWindow(title,
color=init_color,
text_color=init_text_color,
debug=init_debug)
self._menu = None
self._multipress_timer = PoolingTimer(
config.getfloat('CONTROLS', 'multi_press_delay'), False)
# Define states of the application
self._machine = StateMachine(self._pm, self._config, self,
self._window)
self._machine.add_state('wait')
self._machine.add_state('choose')
self._machine.add_state('chosen')
self._machine.add_state('preview')
self._machine.add_state('capture')
self._machine.add_state('processing')
self._machine.add_state('filter')
self._machine.add_state('print')
self._machine.add_state('finish')
# ---------------------------------------------------------------------
# Variables shared with plugins
# Change them may break plugins compatibility
self.capture_nbr = None
self.capture_date = None
self.capture_choices = (4, 1)
self.previous_picture = None
self.previous_animated = None
self.previous_picture_file = None
self.count = Counters(self._config.join_path("counters.pickle"),
taken=0,
printed=0,
forgotten=0,
remaining_duplicates=self._config.getint(
'PRINTER', 'max_duplicates'))
self.camera = camera.get_camera(
config.getint('CAMERA', 'iso'),
config.gettyped('CAMERA', 'resolution'),
config.getint('CAMERA', 'rotation'),
config.getboolean('CAMERA', 'flip'),
config.getboolean('CAMERA', 'delete_internal_memory'))
self.buttons = ButtonBoard(
capture="BOARD" + config.get('CONTROLS', 'picture_btn_pin'),
printer="BOARD" + config.get('CONTROLS', 'print_btn_pin'),
hold_time=config.getfloat('CONTROLS', 'debounce_delay'),
pull_up=True)
self.buttons.capture.when_held = self._on_button_capture_held
self.buttons.printer.when_held = self._on_button_printer_held
self.leds = LEDBoard(
capture="BOARD" + config.get('CONTROLS', 'picture_led_pin'),
printer="BOARD" + config.get('CONTROLS', 'print_led_pin'))
self.printer = Printer(config.get('PRINTER', 'printer_name'),
config.getint('PRINTER', 'max_pages'),
self.count)
def counters(tmpdir):
return Counters(str(tmpdir.join('data.pickle')), nbr_printed=0)