j5j#dZddlmZddlmZddlmZddlmZm Z m Z m Z m Z m Z ddlmZGddeZGd d ZGd d Zy )zDefines progress bar API.) annotations)Protocol)Console) BarColumnMofNCompleteColumnProgressTaskProgressColumn TextColumnTimeRemainingColumn)ColumncHeZdZdZdd d dZdZdZd ddd d Zy) ProgressBaraThe protocol that OCRmyPDF expects progress bar classes to be compatible with. In practice this could be used for any time of monitoring, not just a progress bar. Calling the class should return a new progress bar object, which is activated with ``__enter__`` and terminated with ``__exit__``. An update method is called whenever the progress bar is updated. Progress bar objects will not be reused; a new one will be created for each group of tasks. The progress bar is held in the main process/thread and not updated by child process/threads. When a child notifies the parent of completed work, the parent updates the progress bar. Progress bars should never write to ``sys.stdout``, or they will corrupt the output if OCRmyPDF writes a PDF to standard output. Note: The type of events that OCRmyPDF reports to a progress bar may change in minor releases. Args: total (int | float | None): The total number of work units expected. If ``None``, the total is unknown. For example, if you are processing pages, this might be the number of pages, or if you are measuring overall progress in percent, this might be 100. desc (str | None): A brief description of the current step (e.g. "Scanning contents", "OCR", "PDF/A conversion"). OCRmyPDF updates this before each major step. unit (str | None): A short label for the type of work being tracked (e.g. "page", "%", "image"). disable (bool): If ``True``, progress updates are suppressed (no output). Defaults to ``False``. **kwargs: Future or extra parameters that OCRmyPDF might pass. Implementations should accept and ignore unrecognized keywords gracefully. Example: A simple plugin implementation could look like this: .. code-block:: python from ocrmypdf.pluginspec import ProgressBar from ocrmypdf import hookimpl class ConsoleProgressBar(ProgressBar): def __init__(self, *, total=None, desc=None, unit=None, disable=False, **kwargs): self.total = total self.desc = desc self.unit = unit self.disable = disable self.current = 0 def __enter__(self): if not self.disable: print(f"Starting {self.desc or 'an OCR task'} " f"(total={self.total} {self.unit})" ) return self def __exit__(self, exc_type, exc_value, traceback): if not self.disable: if exc_type is None: print("Completed successfully.") else: print(f"Task ended with error: {exc_value}") return False # Let OCRmyPDF raise any exceptions def update(self, n=1, *, completed=None): if completed is not None: # If 'completed' is given, set self.current # but let's just read it to show usage print(f"Absolute completion reported: {completed}") # Otherwise, we increment by 'n' self.current += n if not self.disable: if self.total: percent = (self.current / self.total) * 100 print( f"{self.desc}: {self.current}" f"/{self.total} ({percent:.1f}%)" ) else: print(f"{self.desc}: {self.current} units done") @hookimpl def get_progressbar_class(): return MyProgressBar F)disablec y)atInitialize a progress bar. This is called once before any work is done. OCRmyPDF supplies the total number of units (or None if unknown), a description of the work, and the type of units. The ``disable`` parameter can be used to turn off progress reporting. Unrecognized keyword arguments should be ignored. Args: total (int | float | None): The total amount of work. If ``None``, the total is unknown. desc (str | None): A description of the current task. May change for different stages. unit (str | None): A short label for the unit of work. disable (bool): If ``True``, no output or logging should be displayed. **kwargs: Extra parameters that may be passed by OCRmyPDF in future versions. N)selftotaldescunitrkwargss K/var/www/html/qr/venv/lib/python3.12/site-packages/ocrmypdf/_progressbar.py__init__zProgressBar.__init__scy)zEnter a progress bar context.Nrrs r __enter__zProgressBar.__enter__rrcy)zExit a progress bar context.Nr)rargss r__exit__zProgressBar.__exit__rrN completedcy)aIncrement the progress bar by ``n`` units, or set an absolute completion. OCRmyPDF calls this method repeatedly while processing pages or other tasks. If your total is known and you track it, you might do something like: .. code-block:: python self.current += n percent = (self.current / total) * 100 The ``completed`` argument can indicate an absolute position, which is particularly helpful if you're tracking a percentage of work (e.g., 0 to 100) and want precise updates. In contrast, the incremental parameter ``n`` is often more useful for page-based increments. Args: n (float, optional): The amount to increment the progress by. Defaults to 1. May be fractional if OCRmyPDF performs partial steps. If you are tracking pages, this is typically how many pages have been processed in the most recent step. completed (float | None, optional): The absolute amount of work completed so far. This can override or supplement the simple increment logic. It's particularly useful for percentage-based tracking (e.g., when ``total`` is 100). Nr)rnr"s rupdatezProgressBar.updaterr)rzint | float | Noner str | Nonerr&rbool)r$floatr" float | None__name__ __module__ __qualname____doc__rrr r%rrrrrsQZD  "       :,+  rrc0eZdZdZdZdZdZddddZy) NullProgressBarz'Progress bar API that takes no actions.c yNr)rrs rrzNullProgressBar.__init__s rc|Sr4rrs rrzNullProgressBar.__enter__s rcyNFrrexc_type exc_value tracebacks rr zNullProgressBar.__exit__srNr!cyr4r)r_argr"s rr%zNullProgressBar.updatesrr4r,rrrr2r2s 1 Trr2cTeZdZdZddddd d dZdZdZd dd d Zy) RichProgressBarz Display progress bar using rich.Ng?F)rr unit_scalerc @d|_ttdtdt t t tf|ddd|d||_||_ |jj|||j||jznd||_ y) NFz([progress.description]{task.description}) min_width) table_columnT)console auto_refreshredirect_stderrredirect_stdoutr)rr) _enteredrr r rr rr progressr@add_task progress_bar)rrErrrr@rrs rrzRichProgressBar.__init__s  :#b1  K    !  !   % MM22  T__%@$//) 3 rcH|jjd|_|S)NT)rJstartrIrs rrzRichProgressBar.__enter__s   rcl|jj|jjyr7)rJrefreshstopr8s rr zRichProgressBar.__exit__s%  rr!c|jsJd|8| |jn|}|jj|j|y|jj|j|y)Nz,Progress bar must be entered before updating)advancer!)rIr@rJr%rL)rr$r"rSs rr%zRichProgressBar.updatesc}}LLL}  )*dooG MM !2!2G D MM !2!2i Hr) rErrstrrr+rr&r@r+rr'r(r,rrrr?r?sm*##&# #  #  #  # !# # J  ItIrr?N)r0 __future__rtypingr rich.consoler rich.progressrrrr r r rich.tabler rr2r?rrrrZsG " Z (Z z   8I8Ir