henklaak/fatamorgana
1
0
Fork 0
forked from jan/fatamorgana
Code Issues Pull Requests Releases Wiki Activity

enable passing in str where an AString or NString is needed.

Browse Source
This commit is contained in:
Jan Petykiewicz 2020-05-19 00:47:17 -07:00
parent 705926d443
commit 99283aaaf0
2 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
fatamorgana/main.py
View File

@ -386,12 +386,12 @@ class Cell:
placements: List[records.Placement] placements: List[records.Placement]
geometry: List[records.geometry_t] geometry: List[records.geometry_t]
def __init__(self, name: Union[NString, int]): def __init__(self, name: Union[NString, str, int]):
""" """
Args: Args:
name: `NString` or "CellName reference" number name: `NString` or "CellName reference" number
""" """
self.name = name self.name = name if isinstance(name, (NString, int)) else NString(name)
self.properties = [] self.properties = []
self.placements = [] self.placements = []
self.geometry = [] self.geometry = []

32
fatamorgana/records.py
View File

@ -921,12 +921,12 @@ class Property(Record):
values (Optional[List[property_value_t]]): List of property values. values (Optional[List[property_value_t]]): List of property values.
is_standard (bool): Whether this is a standard property. is_standard (bool): Whether this is a standard property.
""" """
name: Union[NString, int, None] = None name: Optional[Union[NString, int]] = None
values: Optional[List[property_value_t]] = None values: Optional[List[property_value_t]] = None
is_standard: Optional[bool] = None is_standard: Optional[bool] = None
def __init__(self, def __init__(self,
name: Union[NString, str, int] = None, name: Union[NString, str, int, None] = None,
values: Optional[List[property_value_t]] = None, values: Optional[List[property_value_t]] = None,
is_standard: Optional[bool] = None): is_standard: Optional[bool] = None):
""" """
@ -938,10 +938,10 @@ class Property(Record):
is_standard: `True` if this is a standard property. `None` to use modal. is_standard: `True` if this is a standard property. `None` to use modal.
Default `None`. Default `None`.
""" """
if isinstance(name, str): if isinstance(name, (NString, int)) or name is None:
self.name = NString(name)
else:
self.name = name self.name = name
else:
self.name = NString(name)
self.values = values self.values = values
self.is_standard = is_standard self.is_standard = is_standard
@ -1254,12 +1254,12 @@ class Cell(Record):
""" """
name: Union[int, NString] name: Union[int, NString]
def __init__(self, name: Union[int, NString]): def __init__(self, name: Union[int, str, NString]):
""" """
Args: Args:
name: `NString`, or an int specifying a `CellName` reference number. name: `NString`, or an int specifying a `CellName` reference number.
""" """
self.name = name self.name = name if isinstance(name, (int, NString)) else NString(name)
def merge_with_modals(self, modals: Modals): def merge_with_modals(self, modals: Modals):
modals.reset() modals.reset()
@ -1316,7 +1316,7 @@ class Placement(Record):
def __init__(self, def __init__(self,
flip: bool, flip: bool,
name: Union[NString, str, int] = None, name: Union[NString, str, int, None] = None,
magnification: Optional[real_t] = None, magnification: Optional[real_t] = None,
angle: Optional[real_t] = None, angle: Optional[real_t] = None,
x: Optional[int] = None, x: Optional[int] = None,
@ -1340,10 +1340,10 @@ class Placement(Record):
self.flip = flip self.flip = flip
self.magnification = magnification self.magnification = magnification
self.angle = angle self.angle = angle
if isinstance(name, str): if isinstance(name, (int, NString)) or name is None:
self.name = NString(name)
else:
self.name = name self.name = name
else:
self.name = NString(name)
def get_name(self) -> Union[NString, int]: def get_name(self) -> Union[NString, int]:
return verify_modal(self.name) # type: ignore return verify_modal(self.name) # type: ignore
@ -1449,7 +1449,7 @@ class Text(Record, GeometryMixin):
y (Optional[int]): y-offset, None means reuse modal y (Optional[int]): y-offset, None means reuse modal
repetition (Optional[repetition_t]): Repetition, if any repetition (Optional[repetition_t]): Repetition, if any
""" """
string: Union[AString, int, None] = None string: Optional[Union[AString, int]] = None
layer: Optional[int] = None layer: Optional[int] = None
datatype: Optional[int] = None datatype: Optional[int] = None
x: Optional[int] = None x: Optional[int] = None
@ -1457,7 +1457,7 @@ class Text(Record, GeometryMixin):
repetition: Optional[repetition_t] = None repetition: Optional[repetition_t] = None
def __init__(self, def __init__(self,
string: Union[AString, str, int] = None, string: Union[AString, str, int, None] = None,
layer: Optional[int] = None, layer: Optional[int] = None,
datatype: Optional[int] = None, datatype: Optional[int] = None,
x: Optional[int] = None, x: Optional[int] = None,
@ -1478,10 +1478,10 @@ class Text(Record, GeometryMixin):
self.x = x self.x = x
self.y = y self.y = y
self.repetition = repetition self.repetition = repetition
if isinstance(string, str): if isinstance(string, (AString, int)) or string is None:
self.string = AString(string)
else:
self.string = string self.string = string
else:
self.string = AString(string)
def get_string(self) -> Union[AString, int]: def get_string(self) -> Union[AString, int]:
return verify_modal(self.string) # type: ignore return verify_modal(self.string) # type: ignore