-
Notifications
You must be signed in to change notification settings - Fork 3
/
line_iterator.py
44 lines (35 loc) · 1.42 KB
/
line_iterator.py
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import re
from typing import List
def split_text(text: str) -> List[str]:
return [x.strip() for x in re.compile(r'===============').split(text)]
class Lines:
captions = ('Название компании', 'Описание вакансии', 'Технологии',
'Офис или удаленка', 'Адрес офиса', 'Занятость', 'Ссылка', 'Зарплатная вилка', 'Контакты')
def __init__(self, values: str):
self.values = split_text(values)
self.index = -1
def __iter__(self):
return self
def __next__(self) -> str:
if len(self.captions) == self.index or len(self.values) == self.index:
raise StopIteration
if self.index == -1:
result = '#вакансия\n'
else:
value = self.value
while value == '': # skip empty lines
self.index += 1
value = self.value
result = f'{self.captions[self.index]}: {value}'
self.index += 1
return result
@property
def value(self) -> str:
def add_hashtag(text: str) -> str:
regex = re.compile(r'\w+')
my_list = regex.findall(text)
return ', '.join([f"#{word}" for word in my_list])
result = self.values[self.index]
if self.index in (2, 3, 5):
result = add_hashtag(result)
return result