105 lines
2.7 KiB
Python
105 lines
2.7 KiB
Python
from utils import choice, rndunicode, choice_percent
|
|
|
|
|
|
class Mutations:
|
|
def __init__(self, strongness=0):
|
|
self.strongness = strongness
|
|
|
|
def mutate_attr(self, attr):
|
|
cases = {
|
|
70: lambda x: x,
|
|
10: self.mutate_attr_newline,
|
|
10: self.mutate_attr_unicode,
|
|
10: self.mutate_attr_unicode_rnd_pos,
|
|
}
|
|
|
|
return choice_percent(cases)(attr)
|
|
|
|
def mutate_value(self, value: str):
|
|
cases = {
|
|
70: lambda x: x,
|
|
20: self.mutate_value_newline,
|
|
10: self.mutate_value_unicode,
|
|
}
|
|
|
|
return choice_percent(cases)(value)
|
|
|
|
def mutate_tag(self, tag):
|
|
cases = {
|
|
70: lambda x: x,
|
|
30: self.mutate_tag_insert_random_pos,
|
|
}
|
|
|
|
return choice_percent(cases)(tag)
|
|
|
|
def mutate_js(self, js: str):
|
|
cases = {
|
|
70: lambda x: x,
|
|
30: self.mutate_js_insert_random_pos,
|
|
}
|
|
|
|
return choice_percent(cases)(js)
|
|
|
|
@staticmethod
|
|
def mutate_attr_newline(attr):
|
|
attr.name = f"{attr.name}\n"
|
|
return attr
|
|
|
|
@staticmethod
|
|
def mutate_attr_unicode(attr):
|
|
attr.name = f"{attr.name}{rndunicode()}"
|
|
return attr
|
|
|
|
@staticmethod
|
|
def mutate_attr_unicode_rnd_pos(attr):
|
|
pos = choice(range(len(attr.name)))
|
|
attr.name = f"{attr.name[:pos]}{rndunicode()}{attr.name[pos:]}"
|
|
return attr
|
|
|
|
@staticmethod
|
|
def mutate_value_newline(attr):
|
|
# find a random position in the value
|
|
# and insert a newline
|
|
pos = choice(range(len(attr.value)))
|
|
attr.value = f"{attr.value[:pos]}\n{attr.value[pos:]}"
|
|
return attr
|
|
|
|
@staticmethod
|
|
def mutate_value_unicode(attr):
|
|
pos = choice(range(len(attr.value)))
|
|
attr.value = f"{attr.value[:pos]}{rndunicode()}{attr.value[pos:]}"
|
|
return attr
|
|
|
|
@staticmethod
|
|
def mutate_tag_insert_random_pos(tag):
|
|
pos = choice(range(0, len(str(tag))))
|
|
|
|
chars = {
|
|
50: lambda: "\n",
|
|
20: lambda: rndunicode(),
|
|
20: lambda: choice([" ", "\t"]),
|
|
10: lambda: choice(["\\", "'", "/", '"']),
|
|
}
|
|
|
|
nmutations = choice(range(1, 4))
|
|
|
|
for _ in range(nmutations):
|
|
tag.rndchrpos[pos] = choice_percent(chars)()
|
|
|
|
return tag
|
|
|
|
@staticmethod
|
|
def mutate_js_insert_random_pos(js: str):
|
|
pos = choice(range(len(js)))
|
|
chars = {
|
|
70: lambda: "\n",
|
|
20: lambda: choice([" ", "\t"]),
|
|
10: lambda: choice(["\\", "'", "/", '"']),
|
|
}
|
|
|
|
nmutations = choice(range(1, 4))
|
|
|
|
for _ in range(nmutations):
|
|
js = f"{js[:pos]}{choice_percent(chars)()}{js[pos:]}"
|
|
return js
|