FilesExpand file tree

 master

/

simplify.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

55 lines (44 loc) · 1.65 KB

 master

/

simplify.py

Copy path

Top

File metadata and controls

• Code
• Blame

55 lines (44 loc) · 1.65 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

"""A module to ease code analysis

This module is here to help source code analysis.

"""

import re

from rope.base import codeanalyze, utils

@utils.cached(7)

def real_code(source):

"""Simplify `source` for analysis

It replaces:

* comments with spaces

* strs with a new str filled with spaces

* implicit and explicit continuations with spaces

* tabs and semicolons with spaces

The resulting code is a lot easier to analyze if we are interested

only in offsets.

"""

collector = codeanalyze.ChangeCollector(source)

for start, end in ignored_regions(source):

if source[start] == '#':

replacement = ' ' * (end - start)

else:

replacement = '"%s"' % (' ' * (end - start - 2))

collector.add_change(start, end, replacement)

source = collector.get_changed() or source

collector = codeanalyze.ChangeCollector(source)

parens = 0

for match in _parens.finditer(source):

i = match.start()

c = match.group()

if c in '({[':

parens += 1

if c in ')}]':

parens -= 1

if c == '\n' and parens > 0:

collector.add_change(i, i + 1, ' ')

source = collector.get_changed() or source

return source.replace('\\\n', ' ').replace('\t', ' ').replace(';', '\n')

@utils.cached(7)

def ignored_regions(source):

"""Return ignored regions like strings and comments in `source` """

return [(match.start(), match.end()) for match in _str.finditer(source)]

_str = re.compile('%s|%s' % (codeanalyze.get_comment_pattern(),

codeanalyze.get_string_pattern()))

_parens = re.compile(r'[\({\[\]}\)\n]')