qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-06-09 12:19:54 +02:00
Code Issues Projects Releases Wiki Activity

counter: allow arbitrary line/region id

Browse source
This commit is contained in:
Konstantin Baierer 2021-03-02 17:41:45 +01:00
parent fffa207658
commit 56b688befe
2 changed files with 16 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

16
qurator/eynollah/utils/counter.py
View file

@ -19,13 +19,25 @@ class EynollahIdCounter():
def set(self, name, val):
self._counter[name] = val
def region_id(self, region_idx=None):
if not region_idx:
region_idx = self._counter['region']
return REGION_ID_TEMPLATE % region_idx
def line_id(self, region_idx=None, line_idx=None):
if not region_idx:
region_idx = self._counter['region']
if not line_idx:
line_idx = self._counter['line']
return LINE_ID_TEMPLATE % (region_idx, line_idx)
@property
def next_region_id(self):
self.inc('region')
self.set('line', 0)
return REGION_ID_TEMPLATE % self._counter['region']
return self.region_id()
@property
def next_line_id(self):
self.inc('line')
return LINE_ID_TEMPLATE % (self._counter['region'], self._counter['line'])
return self.line_id()

2
tests/test_counter.py
View file

@ -8,6 +8,8 @@ def test_counter_string():
assert c.next_line_id == 'region_0002_line_0001'
assert c.next_region_id == 'region_0003'
assert c.next_line_id == 'region_0003_line_0001'
assert c.region_id(999) == 'region_0999'
assert c.line_id(999, 888) == 'region_0999_line_0888'
def test_counter_init():
c = EynollahIdCounter(region_idx=2)