File size: 1,001 Bytes
ce8f04a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
import os
import glob

sources_dir = '/home/bogmaz/PROGRAMY/DOTACJA/grantforge-ai-main/backend/core/search/sources'
files = glob.glob(os.path.join(sources_dir, '*_source.py'))

for fpath in files:
    with open(fpath, 'r', encoding='utf-8') as f:
        lines = f.readlines()
    
    out_lines = []
    in_fallback = False
    
    for line in lines:
        if line.strip().startswith('def _get_verified_fallback'):
            in_fallback = True
            out_lines.append(line)
            out_lines.append('        return []\n')
            continue
            
        if in_fallback:
            # wait until next method definition or end of class
            if line.startswith('    def ') or line.startswith('class '):
                in_fallback = False
                out_lines.append(line)
        else:
            out_lines.append(line)
            
    with open(fpath, 'w', encoding='utf-8') as f:
        f.writelines(out_lines)

print(f'Wiped fallbacks in {len(files)} files.')