def __init__(self, *args, **kwargs):
"""
:param str locale: Current locale.
"""
super().__init__(*args, **kwargs)
self.__inet = Internet()
self.__text = Text('en')
def __init__(self, *args, **kwargs) -> None:
"""Initialize attributes.
:param locale: Current locale.
:param seed: Seed.
"""
super().__init__(*args, **kwargs)
self.__inet = Internet('en', seed=self.seed)
self.__text = Text('en', seed=self.seed)
def __init__(self, *args, **kwargs):
"""Initialize attributes lazily.
:param args: Arguments.
:param kwargs: Keyword arguments.
"""
super().__init__(*args, **kwargs)
self._person = Person
self._address = Address
self._datetime = Datetime
self._business = Business
self._text = Text
self._food = Food
self._science = Science
self._code = Code
self._transport = Transport
self.unit_system = UnitSystem(seed=self.seed)
self.file = File(seed=self.seed)
self.numbers = Numbers(seed=self.seed)
self.development = Development(seed=self.seed)
self.hardware = Hardware(seed=self.seed)
self.clothing_size = ClothingSize(seed=self.seed)
self.internet = Internet(seed=self.seed)
self.path = Path(seed=self.seed)
self.payment = Payment(seed=self.seed)
self.games = Games(seed=self.seed)
self.cryptographic = Cryptographic(seed=self.seed)
self.structure = Structure(seed=self.seed)
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self._personal = Personal
self._address = Address
self._datetime = Datetime
self._business = Business
self._text = Text
self._food = Food
self._science = Science
self._code = Code
self.unit_system = UnitSystem()
self.file = File()
self.numbers = Numbers()
self.development = Development()
self.hardware = Hardware()
self.clothing_sizes = ClothingSizes()
self.internet = Internet()
self.transport = Transport()
self.path = Path()
self.games = Games()
self.cryptographic = Cryptographic()
class Structure(BaseDataProvider):
"""Class for generating structured data."""
def __init__(self, *args, **kwargs) -> None:
"""Initialize attributes.
:param locale: Current locale.
:param seed: Seed.
"""
super().__init__(*args, **kwargs)
self.__inet = Internet('en', seed=self.seed)
self.__text = Text('en', seed=self.seed)
def css(self) -> str:
"""Generate a random snippet of CSS.
:return: CSS.
"""
selector = self.random.choice(CSS_SELECTORS)
css_sel = '{}{}'.format(selector, self.__text.word())
cont_tag = self.random.choice(list(HTML_CONTAINER_TAGS.keys()))
mrk_tag = self.random.choice(HTML_MARKUP_TAGS)
base = '{}'.format(self.random.choice([cont_tag, mrk_tag, css_sel]))
props = '; '.join(
[self.css_property() for _ in range(self.random.randint(1, 6))])
return '{} {{{}}}'.format(base, props)
def css_property(self) -> str:
"""Generate a random snippet of CSS that assigns value to a property.
:return: CSS property.
:Examples:
'background-color: #f4d3a1'
"""
prop = self.random.choice(list(CSS_PROPERTIES.keys()))
val = CSS_PROPERTIES[prop]
if isinstance(val, list):
val = self.random.choice(val)
elif val == 'color':
val = self.__text.hex_color()
elif val == 'size':
val = '{}{}'.format(self.random.randint(1, 99),
self.random.choice(CSS_SIZE_UNITS))
return '{}: {}'.format(prop, val)
def html(self) -> str:
"""Generate a random HTML tag with text inside and some attrs set.
:return: HTML.
:Examples:
'<span class="select" id="careers">
Ports are created with the built-in function open_port.
</span>'
"""
tag_name = self.random.choice(list(HTML_CONTAINER_TAGS))
tag_attributes = list(HTML_CONTAINER_TAGS[tag_name]) # type: ignore
k = self.random.randint(1, len(tag_attributes))
selected_attrs = self.random.sample(tag_attributes, k=k)
attrs = []
for attr in selected_attrs:
attrs.append('{}="{}"'.format(
attr, self.html_attribute_value(tag_name, attr)))
html_result = '<{tag} {attrs}>{content}</{tag}>'
return html_result.format(
tag=tag_name,
attrs=' '.join(attrs),
content=self.__text.sentence(),
)
def html_attribute_value(self,
tag: str = None,
attribute: str = None) -> str:
"""Generate random value for specified HTML tag attribute.
:param tag: An HTML tag.
:param attribute: An attribute of the specified tag.
:return: An attribute.
:raises NotImplementedError: if tag is unsupported.
"""
if not tag:
tag = self.random.choice(list(HTML_CONTAINER_TAGS.keys()), )
if not attribute:
attribute = self.random.choice(
list(HTML_CONTAINER_TAGS[tag]), # type: ignore
)
try:
value = HTML_CONTAINER_TAGS[tag][attribute] # type: ignore
except KeyError:
raise NotImplementedError(
'Tag {} or attribute {} is not supported'.format(
tag, attribute))
if isinstance(value, list):
value = self.random.choice(value)
elif value == 'css':
value = self.css_property()
elif value == 'word':
value = self.__text.word()
elif value == 'url':
value = self.__inet.home_page()
else:
raise NotImplementedError(
'Attribute type {} is not implemented'.format(value))
return value
def json(self,
items: int = 5,
max_depth: int = 3,
recursive: bool = False) -> str:
"""Generate a random snippet of JSON.
:param items: Number of top-level items to produce.
:param max_depth: Maximum depth of each top-level item.
:param recursive:
When used recursively, will return a Python object instead of JSON
string.
:return: JSON
"""
root = self.random.choice([list, dict])()
for _ in range(items):
key = self.__text.word()
if max_depth > 0:
json_values = {
'str':
lambda: self.__text.sentence(),
'int':
lambda: self.random.randint(1, 10000),
'float':
lambda: self.random.random(),
'bool':
lambda: self.random.choice([True, False]),
'null':
lambda: None,
'json':
lambda: self.json(
max_depth=max_depth - 1,
recursive=True,
),
}
vkey = self.random.choice(list(json_values.keys()))
value = json_values[vkey]()
if isinstance(root, list):
root.append(value)
elif isinstance(root, dict):
root[key] = value
if recursive:
return root
return json.dumps(root, indent=4)
class Structure(BaseDataProvider):
"""Class for generating structured data."""
def __init__(self, *args, **kwargs) -> None:
"""Initialize attributes.
:param locale: Current locale.
:param seed: Seed.
"""
super().__init__(*args, **kwargs)
self.__inet = Internet(seed=self.seed)
self.__text = Text('en', seed=self.seed)
def css(self) -> str:
"""Generate a random snippet of CSS.
:return: CSS.
"""
selector = self.random.choice(CSS_SELECTORS)
css_sel = '{}{}'.format(selector, self.__text.word())
cont_tag = self.random.choice(list(HTML_CONTAINER_TAGS.keys()))
mrk_tag = self.random.choice(HTML_MARKUP_TAGS)
base = '{}'.format(self.random.choice([cont_tag, mrk_tag, css_sel]))
props = '; '.join(
[self.css_property() for _ in range(self.random.randint(1, 6))])
return '{} {{{}}}'.format(base, props)
def css_property(self) -> str:
"""Generate a random snippet of CSS that assigns value to a property.
:return: CSS property.
:Examples:
'background-color: #f4d3a1'
"""
prop = self.random.choice(list(CSS_PROPERTIES.keys()))
val = CSS_PROPERTIES[prop]
if isinstance(val, list):
val = self.random.choice(val)
elif val == 'color':
val = self.__text.hex_color()
elif val == 'size':
val = '{}{}'.format(self.random.randint(1, 99),
self.random.choice(CSS_SIZE_UNITS))
return '{}: {}'.format(prop, val)
def html(self) -> str:
"""Generate a random HTML tag with text inside and some attrs set.
:return: HTML.
:Examples:
'<span class="select" id="careers">
Ports are created with the built-in function open_port.
</span>'
"""
tag_name = self.random.choice(list(HTML_CONTAINER_TAGS))
tag_attributes = list(HTML_CONTAINER_TAGS[tag_name]) # type: ignore
k = self.random.randint(1, len(tag_attributes))
selected_attrs = self.random.sample(tag_attributes, k=k)
attrs = []
for attr in selected_attrs:
attrs.append('{}="{}"'.format(
attr, self.html_attribute_value(tag_name, attr)))
html_result = '<{tag} {attrs}>{content}</{tag}>'
return html_result.format(
tag=tag_name,
attrs=' '.join(attrs),
content=self.__text.sentence(),
)
def html_attribute_value(self,
tag: str = None,
attribute: str = None) -> str:
"""Generate random value for specified HTML tag attribute.
:param tag: An HTML tag.
:param attribute: An attribute of the specified tag.
:return: An attribute.
:raises NotImplementedError: if tag is unsupported.
"""
if not tag:
tag = self.random.choice(list(HTML_CONTAINER_TAGS.keys()), )
if not attribute:
attribute = self.random.choice(
list(HTML_CONTAINER_TAGS[tag]), # type: ignore
)
try:
value = HTML_CONTAINER_TAGS[tag][attribute] # type: ignore
except KeyError:
raise NotImplementedError(
'Tag {} or attribute {} is not supported'.format(
tag, attribute))
if isinstance(value, list):
value = self.random.choice(value)
elif value == 'css':
value = self.css_property()
elif value == 'word':
value = self.__text.word()
elif value == 'url':
value = self.__inet.home_page()
else:
raise NotImplementedError(
'Attribute type {} is not implemented'.format(value))
return value
