refactor python modules

1 files changed, 17 insertions, 0 deletions

diff --git a/python/fatcat_tools/fcid.py b/python/fatcat_tools/fcid.py
new file mode 100644
index 00000000..dd72b242
--- /dev/null
+++ b/python/fatcat_tools/fcid.py
@@ -0,0 +1,17 @@
+
+import base64
+import uuid
+
+def fcid2uuid(s):
+    s = s.split('_')[-1].upper().encode('utf-8')
+    assert len(s) == 26
+    raw = base64.b32decode(s + b"======")
+    return str(uuid.UUID(bytes=raw)).lower()
+
+def uuid2fcid(s):
+    raw = uuid.UUID(s).bytes
+    return base64.b32encode(raw)[:26].lower().decode('utf-8')
+
+def test_fcid():
+    test_uuid = '00000000-0000-0000-3333-000000000001'
+    assert test_uuid == fcid2uuid(uuid2fcid(test_uuid))