Skip to content

Commit 6d252c6

Browse files
Lucas Pololucaspolo
authored andcommitted
Hide number if it is a asterisk
1 parent c9d1fcc commit 6d252c6

File tree

2 files changed

+16
-6
lines changed

2 files changed

+16
-6
lines changed

correios/models/address.py

Lines changed: 15 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -284,13 +284,21 @@ def zip_code_display(self) -> str:
284284

285285
@property
286286
def basic_address(self) -> str:
287-
number = self.number
287+
number = ""
288+
289+
if self.number:
290+
number = ", {}".format(self.number)
291+
288292
if self.complement:
289-
number = "{} - {}".format(self.number, self.complement)
293+
number = "{} - {}".format(number, self.complement)
290294

291295
if self.neighborhood:
292-
return capitalize_phrase("{}, {}, {}".format(self.street, number, self.neighborhood))
293-
return capitalize_phrase("{}, {}".format(self.street, number))
296+
return capitalize_phrase("{}{}, {}".format(self.street, number, self.neighborhood))
297+
298+
if number:
299+
return capitalize_phrase("{}{}".format(self.street, number))
300+
301+
return capitalize_phrase("{}".format(self.street))
294302

295303
@property
296304
def label_address(self) -> str:
@@ -321,7 +329,9 @@ def filtered_number(self) -> str:
321329

322330
@property
323331
def number(self) -> str:
324-
return self.filtered_number or "S/N"
332+
number = self.filtered_number or "S/N"
333+
334+
return number.replace('*', '')
325335

326336
@property
327337
def zip_complement(self) -> str:

tests/test_address_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -388,7 +388,7 @@ def test_basic_address_with_neighborhood_without_complement():
388388
('km 5', '5', '5', '5'),
389389
('s/n', '', 'S/N', '0'),
390390
('S/N', '', 'S/N', '0'),
391-
('*', '*', '*', '*'),
391+
('*', '*', '', '*'),
392392
))
393393
def test_address_number_handling(raw, filtered, number, zip_complement):
394394
address = Address(

0 commit comments

Comments
 (0)