mirror of
https://github.com/mihakralj/QuanTAlib.git
synced 2026-07-28 09:47:43 +00:00
Miha Kralj
29 lines
1.1 KiB
Python
29 lines
1.1 KiB
Python
import re, os
|
|
|
|
# Collect all function names from new category files
|
|
new_fns = set()
|
|
for f in sorted(os.listdir('python/quantalib')):
|
|
if f.endswith('.py') and not f.startswith('_') and f != 'indicators.py':
|
|
content = open(f'python/quantalib/{f}', encoding='utf-8').read()
|
|
new_fns.update(re.findall(r'^def (\w+)\(', content, re.MULTILINE))
|
|
|
|
# Collect from old indicators.py
|
|
old_content = open('python/quantalib/indicators.py', encoding='utf-8').read()
|
|
old_fns = set(re.findall(r'^def (\w+)\(', old_content, re.MULTILINE))
|
|
old_fns = {f for f in old_fns if not f.startswith('_')}
|
|
|
|
missing = sorted(old_fns - new_fns)
|
|
extra = sorted(new_fns - old_fns)
|
|
|
|
with open('python/tools/diff_report.txt', 'w') as out:
|
|
out.write(f'Old indicators.py: {len(old_fns)} public functions\n')
|
|
out.write(f'New category files: {len(new_fns)} functions\n\n')
|
|
out.write(f'Missing from new ({len(missing)}):\n')
|
|
for m in missing:
|
|
out.write(f' {m}\n')
|
|
out.write(f'\nNew indicators not in old ({len(extra)}):\n')
|
|
for e in extra:
|
|
out.write(f' {e}\n')
|
|
|
|
print('Done - see python/tools/diff_report.txt')
|