aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--fuzzycat/fatcat/main.py16
-rw-r--r--fuzzycat/fatcat/matching.py6
-rw-r--r--tests/test_matching.py3
3 files changed, 18 insertions, 7 deletions
diff --git a/fuzzycat/fatcat/main.py b/fuzzycat/fatcat/main.py
index 9113d8f..805e69e 100644
--- a/fuzzycat/fatcat/main.py
+++ b/fuzzycat/fatcat/main.py
@@ -1,8 +1,20 @@
# coding: utf-8
-
"""
Command line entry point for ad-hoc testing.
"""
+from fatcat_openapi_client import ReleaseEntity, ReleaseExtIds
+from fuzzycat.fatcat.matching import match_release_fuzzy
+import argparse
+
+
def main():
- print("fuzzycat fatcat matching") \ No newline at end of file
+ parser = argparse.ArgumentParser()
+ parser.add_argument("-R", "--release", help="match release", action="store_true")
+ parser.add_argument("-t", "--title", help="title")
+
+ args = parser.parse_args()
+
+ if args.release and args.title:
+ re = ReleaseEntity(title=args.title, ext_ids=ReleaseExtIds())
+ print(match_release_fuzzy(re, es="https://search.fatcat.wiki"))
diff --git a/fuzzycat/fatcat/matching.py b/fuzzycat/fatcat/matching.py
index 3ce8269..ba0fef5 100644
--- a/fuzzycat/fatcat/matching.py
+++ b/fuzzycat/fatcat/matching.py
@@ -239,7 +239,7 @@ def verify_container_name(a: str, b: str) -> MatchStatus:
status = verify_serial_name(a, b)
if status != MatchStatus.AMBIGIOUS:
return status
-
+
# TODO: add additional verification, string match and common patterns.
@@ -269,5 +269,5 @@ def verify_release_match(a: ReleaseEntity, b: ReleaseEntity) -> MatchStatus:
return MatchStatus.DIFFERENT
if cmp_result["hits"] < cmp_result["misses"]:
return MatchStatus.AMBIGIOUS
-
- # TODO: do title verification, apply string cleanups, etc. \ No newline at end of file
+
+ # TODO: do title verification, apply string cleanups, etc.
diff --git a/tests/test_matching.py b/tests/test_matching.py
index dcaa6bf..6ae393b 100644
--- a/tests/test_matching.py
+++ b/tests/test_matching.py
@@ -1,5 +1,4 @@
# coding: utf-8
-
"""
Test cases for fuzzy matching.
-""" \ No newline at end of file
+"""