Revision e9c73313 kamaki/cli/argument/__init__.py

b/kamaki/cli/argument/__init__.py
332 332
        newarg._value = self._value
333 333
        return newarg
334 334

  
335
    def get_generator(self, message, message_len=25):
335
    def get_generator(self, message, message_len=25, timeout=False):
336 336
        """Get a generator to handle progress of the bar (gen.next())"""
337 337
        if self.value:
338 338
            return None
......
341 341
        except NameError:
342 342
            self.value = None
343 343
            return self.value
344
        if timeout:
345
            bar_phases = list(self.bar.phases)
346
            bar_phases[0], bar_phases[-1] = bar_phases[-1], ''
347
            self.bar.phases = bar_phases
348
            self.bar.empty_fill = bar_phases[0]
349
            self.bar.bar_prefix = ' (Timeout:'
350
            self.bar.bar_suffix = ' '
351
            self.bar.suffix = '%(eta)ds)'
352
            self.bar.eta = 120
353
        else:
354
            self.bar.suffix = '%(percent)d%% - %(eta)ds'
344 355
        self.bar.message = message.ljust(message_len)
345
        self.bar.suffix = '%(percent)d%% - %(eta)ds'
346 356
        self.bar.start()
347 357

  
348 358
        def progress_gen(n):

Also available in: Unified diff