@@ -29,7 +29,7 @@ if __name__ == "__main__":
2929}
3030
3131fn save_main_files ( project_info : & ProjectInfo ) -> Result < ( ) > {
32- let module = project_info. source_dir . replace ( '-' , "_" ) ;
32+ let module = project_info. source_dir . replace ( [ ' ' , '-' ] , "_" ) ;
3333 let src = project_info. base_dir ( ) . join ( & module) ;
3434 let main = src. join ( "main.py" ) ;
3535 let main_content = create_main_file ( ) ;
@@ -56,7 +56,7 @@ def test_main():
5656}
5757
5858fn save_main_test_file ( project_info : & ProjectInfo ) -> Result < ( ) > {
59- let module = project_info. source_dir . replace ( '-' , "_" ) ;
59+ let module = project_info. source_dir . replace ( [ ' ' , '-' ] , "_" ) ;
6060 let file_path = project_info. base_dir ( ) . join ( "tests/test_main.py" ) ;
6161 let content = create_main_test_file ( & module) ;
6262
@@ -77,7 +77,7 @@ def test_sum_as_string():
7777}
7878
7979fn save_pyo3_test_file ( project_info : & ProjectInfo ) -> Result < ( ) > {
80- let module = project_info. source_dir . replace ( '-' , "_" ) ;
80+ let module = project_info. source_dir . replace ( [ ' ' , '-' ] , "_" ) ;
8181 let file_path = project_info
8282 . base_dir ( )
8383 . join ( format ! ( "tests/test_{}.py" , & module) ) ;
@@ -148,7 +148,7 @@ fn save_test_init_file(project_info: &ProjectInfo) -> Result<()> {
148148}
149149
150150fn save_project_init_file ( project_info : & ProjectInfo ) -> Result < ( ) > {
151- let module = project_info. source_dir . replace ( '-' , "_" ) ;
151+ let module = project_info. source_dir . replace ( [ ' ' , '-' ] , "_" ) ;
152152 let file_path = project_info
153153 . base_dir ( )
154154 . join ( format ! ( "{}/__init__.py" , & module) ) ;
@@ -168,7 +168,7 @@ def sum_as_string(a: int, b: int) -> str: ...
168168}
169169
170170pub fn save_pyi_file ( project_info : & ProjectInfo ) -> Result < ( ) > {
171- let module = project_info. source_dir . replace ( '-' , "_" ) ;
171+ let module = project_info. source_dir . replace ( [ ' ' , '-' ] , "_" ) ;
172172 let file_path = project_info
173173 . base_dir ( )
174174 . join ( format ! ( "{}/_{}.pyi" , & module, & module) ) ;
@@ -184,7 +184,7 @@ fn create_version_file(version: &str) -> String {
184184}
185185
186186fn save_version_file ( project_info : & ProjectInfo ) -> Result < ( ) > {
187- let module = project_info. source_dir . replace ( '-' , "_" ) ;
187+ let module = project_info. source_dir . replace ( [ ' ' , '-' ] , "_" ) ;
188188 let file_path = project_info
189189 . base_dir ( )
190190 . join ( format ! ( "{}/_version.py" , & module) ) ;
@@ -244,7 +244,7 @@ else:
244244}
245245
246246fn save_version_test_file ( project_info : & ProjectInfo ) -> Result < ( ) > {
247- let module = project_info. source_dir . replace ( '-' , "_" ) ;
247+ let module = project_info. source_dir . replace ( [ ' ' , '-' ] , "_" ) ;
248248 let file_path = project_info. base_dir ( ) . join ( "tests/test_version.py" ) ;
249249 let content = create_version_test_file ( & module, & project_info. project_manager ) ;
250250 save_file_with_content ( & file_path, & content) ?;
0 commit comments