summaryrefslogtreecommitdiffstats
path: root/python/fatcat_tools/fcid.py
blob: 07463f626b081fd49f662062949dd8da96726ae7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
import base64
import uuid


def fcid2uuid(fcid: str) -> str:
    """
    Converts a fatcat identifier (base32 encoded string) to a uuid.UUID object
    """
    b = fcid.split("_")[-1].upper().encode("utf-8")
    assert len(b) == 26
    raw_bytes = base64.b32decode(b + b"======")
    return str(uuid.UUID(bytes=raw_bytes)).lower()


def uuid2fcid(s: str) -> str:
    """
    Converts a uuid.UUID object to a fatcat identifier (base32 encoded string)
    """
    raw = uuid.UUID(s).bytes
    return base64.b32encode(raw)[:26].lower().decode("utf-8")


def test_fcid() -> None:
    test_uuid = "00000000-0000-0000-3333-000000000001"
    assert test_uuid == fcid2uuid(uuid2fcid(test_uuid))