Merge pull request #32 from online-judge-tools/feature/generator-range

Use type infomation to generate generate.py
online-judge-tools · Jun 19, 2020 · 77ce848 · 77ce848
2 parents e03df1e + 507abe4
commit 77ce848
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/onlinejudge_template/generator/python.py b/onlinejudge_template/generator/python.py
@@ -105,8 +105,20 @@ def _generate_input_dfs(node: FormatNode, *, declared: Set[str], initialized: Se
     # traverse AST
     if isinstance(node, ItemNode):
         var = _get_variable(decl=decls[node.name], indices=node.indices)
+        type_ = decls[node.name].type
         initialized.add(node.name)
-        return OtherNode(line=f"""{var} = random.randint(1, 10 ** 9)  # TODO: edit here""")
+        if type_ == VarType.IndexInt:
+            return OtherNode(line=f"""{var} = random.randint(1, 1000)  # TODO: edit here""")
+        elif type_ == VarType.ValueInt:
+            return OtherNode(line=f"""{var} = random.randint(1, 10 ** 9)  # TODO: edit here""")
+        elif type_ == VarType.Float:
+            return OtherNode(line=f"""{var} = 100.0 * random.random()  # TODO: edit here""")
+        elif type_ == VarType.String:
+            return OtherNode(line=f"""{var} = ''.join([random.choice('abcde') for range(random.randint(1, 100))])  # TODO: edit here""")
+        elif type_ == VarType.Char:
+            return OtherNode(line=f"""{var} = random.choice('abcde')  # TODO: edit here""")
+        else:
+            return OtherNode(line=f"""{var} = None  # TODO: edit here""")
     elif isinstance(node, NewlineNode):
         return SentencesNode(sentences=[])
     elif isinstance(node, SequenceNode):