| 155 | |
| 156 | |
| 157 | class CommentRemover(ast.NodeTransformer): |
| 158 | def __init__(self, code): |
| 159 | self.root = ast.parse(code) |
| 160 | |
| 161 | def visit_Import(self, node): |
| 162 | return None |
| 163 | |
| 164 | def visit_ImportFrom(self, node): |
| 165 | return None |
| 166 | |
| 167 | def visit_Expr(self, node): |
| 168 | self.generic_visit(node) |
| 169 | if type(node.value) == ast.Constant and isinstance(node.value.value, str): |
| 170 | return None |
| 171 | else: |
| 172 | return node |
| 173 | |
| 174 | def run(self): |
| 175 | self.visit(self.root) |
| 176 | placeholder = PlaceHolder() |
| 177 | self.root = placeholder.run(self.root) |
| 178 | ast.fix_missing_locations(self.root) |
| 179 | new_code = ast.unparse(self.root) |
| 180 | return new_code |
| 181 | |
| 182 | |
| 183 | |