mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 05:53:43 +00:00
ea26372083
Normal hipchat exports use integer ids for their users and "rooms," which we just borrowed during conversion. Atlassian Stride uses stride UUIDs for these instead, but otherwise has the same export format. We now introduce IdMapper to handle external ids that aren't integer. The IdMapper will map UUID ids to ints and remember them. For ints it just leaves them alone. Fixes #10805.
82 lines
1.8 KiB
Python
82 lines
1.8 KiB
Python
from typing import Any, Callable, Dict
|
|
|
|
'''
|
|
This module helps you set up a bunch
|
|
of sequences, similar to how database
|
|
sequences work.
|
|
|
|
You need to be a bit careful here, since
|
|
you're dealing with a big singleton, but
|
|
for data imports that's usually easy to
|
|
manage. See hipchat.py for example usage.
|
|
'''
|
|
|
|
def _seq() -> Callable[[], int]:
|
|
i = 0
|
|
|
|
def next_one() -> int:
|
|
nonlocal i
|
|
i += 1
|
|
return i
|
|
|
|
return next_one
|
|
|
|
def sequencer() -> Callable[[str], int]:
|
|
'''
|
|
Use like this:
|
|
|
|
NEXT_ID = sequencer()
|
|
message_id = NEXT_ID('message')
|
|
'''
|
|
seq_dict = dict() # type: Dict[str, Callable[[], int]]
|
|
|
|
def next_one(name: str) -> int:
|
|
if name not in seq_dict:
|
|
seq_dict[name] = _seq()
|
|
seq = seq_dict[name]
|
|
return seq()
|
|
|
|
return next_one
|
|
|
|
'''
|
|
NEXT_ID is a singleton used by an entire process, which is
|
|
almost always reasonable. If you want to have two parallel
|
|
sequences, just use different `name` values.
|
|
|
|
This object gets created once and only once during the first
|
|
import of the file.
|
|
'''
|
|
|
|
NEXT_ID = sequencer()
|
|
|
|
def is_int(key: Any) -> bool:
|
|
try:
|
|
n = int(key)
|
|
except ValueError:
|
|
return False
|
|
|
|
return n <= 999999999
|
|
|
|
class IdMapper:
|
|
def __init__(self) -> None:
|
|
self.map = dict() # type: Dict[Any, int]
|
|
self.cnt = 0
|
|
|
|
def has(self, their_id: Any) -> bool:
|
|
return their_id in self.map
|
|
|
|
def get(self, their_id: Any) -> int:
|
|
if their_id in self.map:
|
|
return self.map[their_id]
|
|
|
|
if is_int(their_id):
|
|
our_id = int(their_id)
|
|
if self.cnt > 0:
|
|
raise Exception('mixed key styles')
|
|
else:
|
|
self.cnt += 1
|
|
our_id = self.cnt
|
|
|
|
self.map[their_id] = our_id
|
|
return our_id
|