Merge pull request #399 from sanders41/future-annotations

Add __future__ annotations to the main files

Author: sanders41

src/python_files.rs

@@ -7,7 +7,9 @@ use crate::project_info::{ProjectInfo, ProjectManager};
 
 fn create_dunder_main_file(module: &str) -> String {
     format!(
-        r#"from {module}.main import main  #  pragma: no cover
+        r#"from __future__ import annotations
+
+from {module}.main import main  #  pragma: no cover
 
 if __name__ == "__main__":
     raise SystemExit(main())
@@ -16,7 +18,10 @@ if __name__ == "__main__":
 }
 
 fn create_main_file() -> String {
-    r#"def main() -> int:
+    r#"from __future__ import annotations
+
+
+def main() -> int:
     print("Hello world!")  # noqa: T201
 
     return 0

src/snapshots/python_project__python_files__tests__save_main_files-2.snap

@@ -2,4 +2,4 @@
 source: src/python_files.rs
 expression: main_content
 ---
-"def main() -> int:\n    print(\"Hello world!\")  # noqa: T201\n\n    return 0\n\n\nif __name__ == \"__main__\":\n    raise SystemExit(main())\n"
+"from __future__ import annotations\n\n\ndef main() -> int:\n    print(\"Hello world!\")  # noqa: T201\n\n    return 0\n\n\nif __name__ == \"__main__\":\n    raise SystemExit(main())\n"

src/snapshots/python_project__python_files__tests__save_main_files.snap

@@ -2,4 +2,4 @@
 source: src/python_files.rs
 expression: dunder_main_content
 ---
-"from my_project.main import main  #  pragma: no cover\n\nif __name__ == \"__main__\":\n    raise SystemExit(main())\n"
+"from __future__ import annotations\n\nfrom my_project.main import main  #  pragma: no cover\n\nif __name__ == \"__main__\":\n    raise SystemExit(main())\n"