aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorTrygve Laugstøl <trygvis@inamo.no>2019-04-30 07:49:14 +0200
committerTrygve Laugstøl <trygvis@inamo.no>2019-04-30 07:49:14 +0200
commit054019f7afb7aeeac680493c0b176eb35a832c96 (patch)
tree0b5693d15e917cd97461daa29c02896f4dbfa7fd /src
parent37e4be24129b6980e3e9fced7345d4a596af3d58 (diff)
downloadee-python-054019f7afb7aeeac680493c0b176eb35a832c96.tar.gz
ee-python-054019f7afb7aeeac680493c0b176eb35a832c96.tar.bz2
ee-python-054019f7afb7aeeac680493c0b176eb35a832c96.tar.xz
ee-python-054019f7afb7aeeac680493c0b176eb35a832c96.zip
part:
o Experimental importer from Excel-like files.
Diffstat (limited to 'src')
-rw-r--r--src/ee/part/excel.py61
-rw-r--r--src/ee/tools/part_from_excel.py27
2 files changed, 88 insertions, 0 deletions
diff --git a/src/ee/part/excel.py b/src/ee/part/excel.py
new file mode 100644
index 0000000..cbbffde
--- /dev/null
+++ b/src/ee/part/excel.py
@@ -0,0 +1,61 @@
+from pathlib import Path
+from typing import Optional, Mapping
+
+from pyexcel import get_book
+
+from ee.money import Money
+from ee.part import Part, PartDb
+from ee.xml import types
+
+__all__ = [
+ "from_excel",
+]
+
+
+def from_excel(path: Path, sheet_name: Optional[str]) -> PartDb:
+ wb = get_book(file_name=str(path))
+
+ db = PartDb()
+
+ sheet = wb.sheet_by_index(0) if sheet_name is None else wb.sheet_by_name(sheet_name)
+
+ sheet = list(sheet)
+
+ header: Mapping[str, int] = {name.lower(): idx for idx, name in enumerate(sheet[0]) if len(name.strip())}
+ print(header)
+
+ mpn_idx = header.get("mpn", None)
+ desc_idx = header.get("description", None)
+ price_idx = header.get("price", None)
+ url_idx = header.get("url", None)
+
+ print("mpn: {}, desc: {}, price: {}, url: {}".format(mpn_idx, desc_idx, price_idx, url_idx))
+
+ for idx, row in enumerate(sheet[1:], 1):
+ mpn = row[mpn_idx] if mpn_idx is not None else ""
+ desc = row[desc_idx].strip() if desc_idx is not None else ""
+ price = row[price_idx] if price_idx is not None else None
+ url = row[url_idx] if url_idx is not None else None
+
+ print(row[desc_idx])
+
+ if len(mpn) == 0 and len(desc) == 0:
+ print("Bad part, line #{}. MPN or description is required".format(idx))
+ continue
+
+ part = Part(types.Part())
+
+ if not mpn:
+ mpn = desc.replace(" ", "-").lower()
+
+ part.add_mpn(mpn)
+ part.xml.descriptionProp = desc
+
+ if price:
+ part.add_price_break(1, Money(price))
+ if url:
+ part.get_links().append(types.Link(url=url, relation="canonical"))
+
+ db.add_entry(part, False)
+
+ return db
diff --git a/src/ee/tools/part_from_excel.py b/src/ee/tools/part_from_excel.py
new file mode 100644
index 0000000..63c7b65
--- /dev/null
+++ b/src/ee/tools/part_from_excel.py
@@ -0,0 +1,27 @@
+import argparse
+from pathlib import Path
+
+from ee.part import save_db
+from ee.part.excel import from_excel
+
+parser = argparse.ArgumentParser()
+
+parser.add_argument("--in",
+ dest="in_path",
+ required=True,
+ metavar="EXCEL")
+
+parser.add_argument("--out",
+ dest="out_path",
+ required=True,
+ metavar="PART DB")
+
+parser.add_argument("--sheet",
+ metavar="NAME")
+
+args = parser.parse_args()
+
+out_path = Path(args.out_path)
+part_db = from_excel(Path(args.in_path), args.sheet)
+print("len={}".format(part_db.size()))
+save_db(out_path, part_db)