Feat: add lot for charges

pdf_oralia/extract_charge.py

@@ -4,10 +4,19 @@ import numpy as np
 import pandas as pd
 
 
+def get_lot(x):
+    """Return lot number from "RECAPITULATIF DES OPERATIONS" """
+    if x[:2].isdigit():
+        return x[:2]
+    if x[:1].isdigit():
+        return "0" + x[:1]
+    if x[:2] == "PC":
+        return "PC"
+    return ""
+
+
 def extract_charge(table):
     """From pdfplumber table extract the charge dataframe"""
-    for l in table:
-        print(l)
     df = (
         pd.DataFrame(table[1:], columns=table[0])
         .replace("", np.nan)
@@ -26,6 +35,8 @@ def extract_charge(table):
         inplace=True,
     )
 
+    df = df.assign(lot=df["RECAPITULATIF DES OPERATIONS"].map(get_lot))
+
     df = df.astype(
         {
             "Débits": "float64",