| 317 | return new_code |
| 318 | |
| 319 | def remove_space_for_codegen(old_code): |
| 320 | new_code = "" |
| 321 | for line in old_code.splitlines(): |
| 322 | if len(line.strip()) == 0: |
| 323 | new_code += "\n" |
| 324 | continue |
| 325 | lspace = len(line) - len(line.lstrip()) |
| 326 | if lspace % 4 == 2: |
| 327 | new_code += " "*(lspace - 2) |
| 328 | elif lspace%4 == 0: |
| 329 | new_code += " "*lspace |
| 330 | items = line.lstrip().split(" ") |
| 331 | for index, i in enumerate(items): |
| 332 | new_code += i.replace(" ", "") |
| 333 | if index < len(items) - 1: |
| 334 | new_code += " " |
| 335 | |
| 336 | new_code += "\n" |
| 337 | |
| 338 | return new_code |
| 339 | |
| 340 | |
| 341 | |