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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
#
# Copyright 2013, 2018 Free Software Foundation, Inc.
#
# This file is part of GNU Radio
#
# SPDX-License-Identifier: GPL-3.0-or-later
#
#
""" Edit C++ files for bindings """
import re
import logging
logger = logging.getLogger(__name__)
class CPPFileEditor(object):
"""A tool for editing CMakeLists.txt files. """
def __init__(self, filename, separator='\n ', indent=' '):
self.filename = filename
with open(filename, 'r') as f:
self.cfile = f.read()
self.separator = separator
self.indent = indent
# def append_value(self, entry, value, to_ignore_start='', to_ignore_end=''):
# """ Add a value to an entry. """
# regexp = re.compile(r'({}\({}[^()]*?)\s*?(\s?{})\)'.format(entry, to_ignore_start, to_ignore_end),
# re.MULTILINE)
# substi = r'\1' + self.separator + value + r'\2)'
# (self.cfile, nsubs) = regexp.subn(substi, self.cfile, count=1)
# return nsubs
def append_value(self, start_tag, end_tag, value):
""" Add a value to an entry. """
cfile_lines = self.cfile.splitlines()
start_line_idx = [cfile_lines.index(s) for s in cfile_lines if start_tag in s][0]
end_line_idx = [cfile_lines.index(s) for s in cfile_lines if end_tag in s][0]
self.cfile = '\n'.join((cfile_lines[0:end_line_idx]+[self.indent + value]+cfile_lines[end_line_idx:]))
return 1
def remove_value(self, start_tag, end_tag, value):
cfile_lines = self.cfile.splitlines()
try:
start_line_idx = [cfile_lines.index(s) for s in cfile_lines if start_tag in s][0]
end_line_idx = [cfile_lines.index(s) for s in cfile_lines if end_tag in s][0]
except:
logger.warning("Could not find start or end tags in search")
return 0
try:
lines_between_tags = cfile_lines[(start_line_idx+1):end_line_idx]
remove_index = [lines_between_tags.index(s) for s in cfile_lines if value in s][0]
lines_between_tags.pop(remove_index)
except:
return 0
self.cfile = '\n'.join((cfile_lines[0:(start_line_idx+1)]+lines_between_tags+cfile_lines[end_line_idx:]))
return 1
def delete_entry(self, entry, value_pattern=''):
"""Remove an entry from the current buffer."""
regexp = r'{}\s*\([^()]*{}[^()]*\)[^\n]*\n'.format(entry, value_pattern)
regexp = re.compile(regexp, re.MULTILINE)
(self.cfile, nsubs) = re.subn(regexp, '', self.cfile, count=1)
return nsubs
def write(self):
""" Write the changes back to the file. """
with open(self.filename, 'w') as f:
f.write(self.cfile)
def remove_double_newlines(self):
"""Simply clear double newlines from the file buffer."""
self.cfile = re.compile('\n\n\n+', re.MULTILINE).sub('\n\n', self.cfile)
def find_filenames_match(self, regex):
""" Find the filenames that match a certain regex
on lines that aren't comments """
filenames = []
reg = re.compile(regex)
fname_re = re.compile(r'[a-zA-Z]\w+\.\w{1,5}$')
for line in self.cfile.splitlines():
if len(line.strip()) == 0 or line.strip()[0] == '#':
continue
for word in re.split('[ /)(\t\n\r\f\v]', line):
if fname_re.match(word) and reg.search(word):
filenames.append(word)
return filenames
def disable_file(self, fname):
""" Comment out a file.
Example:
add_library(
file1.cc
)
Here, file1.cc becomes #file1.cc with disable_file('file1.cc').
"""
starts_line = False
for line in self.cfile.splitlines():
if len(line.strip()) == 0 or line.strip()[0] == '#':
continue
if re.search(r'\b'+fname+r'\b', line):
if re.match(fname, line.lstrip()):
starts_line = True
break
comment_out_re = r'#\1' + '\n' + self.indent
if not starts_line:
comment_out_re = r'\n' + self.indent + comment_out_re
(self.cfile, nsubs) = re.subn(r'(\b'+fname+r'\b)\s*', comment_out_re, self.cfile)
if nsubs == 0:
logger.warning("Warning: A replacement failed when commenting out {}. Check the CMakeFile.txt manually.".format(fname))
elif nsubs > 1:
logger.warning("Warning: Replaced {} {} times (instead of once). Check the CMakeFile.txt manually.".format(fname, nsubs))
def comment_out_lines(self, pattern, comment_str='#'):
""" Comments out all lines that match with pattern """
for line in self.cfile.splitlines():
if re.search(pattern, line):
self.cfile = self.cfile.replace(line, comment_str+line)
def check_for_glob(self, globstr):
""" Returns true if a glob as in globstr is found in the cmake file """
glob_re = r'GLOB\s[a-z_]+\s"{}"'.format(globstr.replace('*', r'\*'))
return re.search(glob_re, self.cfile, flags=re.MULTILINE|re.IGNORECASE) is not None
|
