Spaces:
Running
Running
| #!/usr/bin/env python3 | |
| """ | |
| DroneKit patcher to fix compatibility with Python 3.10+ | |
| This script needs to be run before importing DroneKit and it will patch | |
| the DroneKit package directly to use collections.abc instead of collections | |
| for MutableMapping. | |
| """ | |
| import os | |
| import sys | |
| import re | |
| import site | |
| from pathlib import Path | |
| def patch_dronekit_files(): | |
| """ | |
| Patch DroneKit Python files to use collections.abc instead of collections | |
| for MutableMapping. | |
| """ | |
| # Find DroneKit install path | |
| try: | |
| import dronekit | |
| print("Error: DroneKit already imported. This script must be run before importing DroneKit.") | |
| return False | |
| except AttributeError: | |
| # This is the error we're trying to fix | |
| pass | |
| # Try to find dronekit in site-packages | |
| dronekit_paths = [] | |
| # Check standard site-packages locations | |
| for site_dir in site.getsitepackages() + [site.getusersitepackages()]: | |
| potential_path = os.path.join(site_dir, "dronekit") | |
| if os.path.exists(potential_path): | |
| dronekit_paths.append(potential_path) | |
| # Also check current directory and its parent | |
| for relative_path in ["dronekit", "../dronekit", "lib/dronekit"]: | |
| if os.path.exists(relative_path): | |
| dronekit_paths.append(os.path.abspath(relative_path)) | |
| if not dronekit_paths: | |
| print("Could not find DroneKit installation. Please install it with pip install dronekit") | |
| return False | |
| patched_files = 0 | |
| for dronekit_path in dronekit_paths: | |
| print(f"Found DroneKit at: {dronekit_path}") | |
| # Find all Python files | |
| for root, _, files in os.walk(dronekit_path): | |
| for file in files: | |
| if file.endswith(".py"): | |
| file_path = os.path.join(root, file) | |
| # Read the file content | |
| with open(file_path, 'r') as f: | |
| content = f.read() | |
| # Check if the file uses collections.MutableMapping | |
| if "collections.MutableMapping" in content or "collections import MutableMapping" in content: | |
| # Replace direct imports | |
| modified_content = re.sub( | |
| r'from collections import (\w+, )*MutableMapping(, \w+)*', | |
| r'from collections.abc import MutableMapping', | |
| content | |
| ) | |
| # Replace usage in code | |
| modified_content = re.sub( | |
| r'collections\.MutableMapping', | |
| r'collections.abc.MutableMapping', | |
| modified_content | |
| ) | |
| # Write the modified content back to the file | |
| with open(file_path, 'w') as f: | |
| f.write(modified_content) | |
| print(f"Patched: {file_path}") | |
| patched_files += 1 | |
| if patched_files > 0: | |
| print(f"Successfully patched {patched_files} DroneKit files to use collections.abc.MutableMapping") | |
| return True | |
| else: | |
| print("No DroneKit files needed patching or files could not be found.") | |
| return False | |
| if __name__ == "__main__": | |
| # Only apply patch for Python 3.10+ | |
| if sys.version_info >= (3, 10): | |
| success = patch_dronekit_files() | |
| if success: | |
| print("DroneKit patched successfully for Python 3.10+ compatibility") | |
| else: | |
| print("Failed to patch DroneKit") | |
| else: | |
| print(f"Python version {sys.version_info.major}.{sys.version_info.minor} does not require patching") |