Antigravity Agent
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
import os
|
|
|
|
def fix_file(filepath):
|
|
with open(filepath, 'r') as f:
|
|
content = f.read()
|
|
|
|
if 'utcnow()' not in content:
|
|
return
|
|
|
|
# Replace utcnow()
|
|
new_content = content.replace('datetime.now(timezone.utc)', 'datetime.now(timezone.utc)')
|
|
|
|
# Add timezone import if missing
|
|
if 'from datetime import timezone' not in new_content and 'import timezone' not in new_content:
|
|
# Try to find a good place to insert it
|
|
lines = new_content.split('\n')
|
|
for i, line in enumerate(lines):
|
|
if line.startswith('from datetime import') or line.startswith('import datetime'):
|
|
lines.insert(i + 1, 'from datetime import timezone')
|
|
break
|
|
else:
|
|
# If no datetime import found, just put it at the top
|
|
lines.insert(0, 'from datetime import timezone')
|
|
new_content = '\n'.join(lines)
|
|
|
|
with open(filepath, 'w') as f:
|
|
f.write(new_content)
|
|
print(f"Fixed {filepath}")
|
|
|
|
def main():
|
|
root_dir = '/tmp/becomingone'
|
|
for dirpath, _, filenames in os.walk(root_dir):
|
|
if '.git' in dirpath:
|
|
continue
|
|
for f in filenames:
|
|
if f.endswith('.py'):
|
|
fix_file(os.path.join(dirpath, f))
|
|
|
|
if __name__ == '__main__':
|
|
main()
|