add class of a city

This commit is contained in:
Pawel Felcyn 2023-03-26 21:29:40 +02:00
parent 8501e7e7b4
commit c41f2a27a5
1 changed files with 39 additions and 0 deletions

39
city.py Normal file
View File

@ -0,0 +1,39 @@
from typing import List, Tuple
from garbageCan import GarbageCan
class Node:
garbageCan: GarbageCan
id: int
def __init__(self, id: int, can: GarbageCan) -> None:
self.id = id
self.can = can
class City:
nodes: List[Node]
streets: List[Tuple[int, int]]
def __init__(self) -> None:
self.nodes = []
self.streets = []
def add_node(self, node: Node) -> None:
self.nodes.append(node)
def add_street(self, street: Tuple[int, int]) -> None:
firstFound: bool = False
secondFound: bool = False
for node in self.nodes:
if firstFound and secondFound:
break
if node.id == street.__getitem__(0):
firstFound = True
continue
if node.id == street.__getitem__(1):
secondFound = True
if not firstFound or not secondFound:
return
self.streets.append(street)