43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
|
import time
|
||
|
|
from typing import Dict, List, Optional
|
||
|
|
|
||
|
|
from .lineage_model import ContributorInfo
|
||
|
|
|
||
|
|
|
||
|
|
class ContributorIndex:
|
||
|
|
def __init__(self):
|
||
|
|
self._contributors: Dict[str, ContributorInfo] = {}
|
||
|
|
|
||
|
|
def register_contributor(self, name: str) -> ContributorInfo:
|
||
|
|
if name not in self._contributors:
|
||
|
|
self._contributors[name] = ContributorInfo(
|
||
|
|
name=name,
|
||
|
|
registered_at=int(time.time() * 1000),
|
||
|
|
contribution_count=1
|
||
|
|
)
|
||
|
|
else:
|
||
|
|
self._contributors[name].contribution_count += 1
|
||
|
|
|
||
|
|
return self._contributors[name]
|
||
|
|
|
||
|
|
def get_contributor(self, name: str) -> Optional[ContributorInfo]:
|
||
|
|
return self._contributors.get(name)
|
||
|
|
|
||
|
|
def list_contributors(self) -> List[ContributorInfo]:
|
||
|
|
return list(self._contributors.values())
|
||
|
|
|
||
|
|
|
||
|
|
_global_index = ContributorIndex()
|
||
|
|
|
||
|
|
|
||
|
|
def register_contributor(name: str) -> ContributorInfo:
|
||
|
|
return _global_index.register_contributor(name)
|
||
|
|
|
||
|
|
|
||
|
|
def get_contributor(name: str) -> Optional[ContributorInfo]:
|
||
|
|
return _global_index.get_contributor(name)
|
||
|
|
|
||
|
|
|
||
|
|
def list_contributors() -> List[ContributorInfo]:
|
||
|
|
return _global_index.list_contributors()
|