Каплан Владимир Сергеевич 611 ВМК SequenceDiv 1275			Пак Алиса, 611 группа SequenceDiv 796
f	1	import sys	f	1	import sys
	2	import ast		2	import ast
	3			3
n	4	class X(ast.NodeTransformer):	n	4	class FloorDivTransformer(ast.NodeTransformer):
	5			5
t	6	def visit_BinOp(self, n):	t	6	def visit_BinOp(self, node):
	7	self.generic_visit(n)		7	self.generic_visit(node)
	8	if isinstance(n.op, ast.FloorDiv):		8	if isinstance(node.op, ast.FloorDiv):
	9	return ast.copy_location(ast.Call(func=ast.Name(id='y', ctx=		9	return ast.copy_location(ast.Call(func=ast.Name(id='floor_di
	>	ast.Load()), args=[n.left, n.right], keywords=[]), n)		>	v', ctx=ast.Load()), args=[node.left, node.right], keywords=[]), node)
	10	return n		10	return node
	11	i = sys.stdin.read()		11	input_code = sys.stdin.read()
	12	t = ast.parse(i)		12	tree = ast.parse(input_code)
	13	x = X()		13	transformer = FloorDivTransformer()
	14	t = x.visit(t)		14	tree = transformer.visit(tree)
	15	ast.fix_missing_locations(t)		15	ast.fix_missing_locations(tree)
	16	y_code = '\ndef y(a, b):\n    if isinstance(a, int):\n        return a /		16	floor_div_code = '\ndef floor_div(a, b):\n    if isinstance(a, int):\n
	>	/ b\n    elif isinstance(a, (list, tuple, str)):\n        return a[: len		>	return a // b\n    elif isinstance(a, (list, tuple, str)):\n
	>	(a) // b]\n'		>	return a[: len(a) // b]\n    else:\n        raise TypeError("Unsupport
				>	ed type for floor_div")\n'
	17	y_ast = ast.parse(y_code)		17	floor_div_ast = ast.parse(floor_div_code)
	18	t.body = y_ast.body + t.body		18	tree.body = floor_div_ast.body + tree.body
	19	exec(compile(t, filename='<ast>', mode='exec'))		19	exec(compile(tree, filename='<ast>', mode='exec'))

Legends
Colors  Added  Changed Deleted	Links (f)irst change (n)ext change (t)op