summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDamien George <damien.p.george@gmail.com>2019-12-28 11:54:49 +1100
committerDamien George <damien.p.george@gmail.com>2019-12-28 11:54:49 +1100
commit269c9a08b6ac248b7bd67d4fa56f70791b5b7a73 (patch)
tree7187327e45c8a73c156ae2b9a36672e76263cf7a
parent1b844e908c3a867be1e54efdb86a04c6a62b2f26 (diff)
unix/modos: Add uos.rename and uos.rmdir.
The existing uos.remove cannot be used to remove directories, instead uos.rmdir is needed. And also provide uos.rename to get a good set of filesystem functionality without requiring additional Python-level os functions (eg using ffi).
-rw-r--r--ports/unix/modos.c26
1 files changed, 26 insertions, 0 deletions
diff --git a/ports/unix/modos.c b/ports/unix/modos.c
index 161918d4d..c3b0c20b2 100644
--- a/ports/unix/modos.c
+++ b/ports/unix/modos.c
@@ -29,6 +29,7 @@
#include <sys/stat.h>
#include <unistd.h>
#include <errno.h>
+#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <dirent.h>
@@ -123,6 +124,29 @@ STATIC mp_obj_t mod_os_remove(mp_obj_t path_in) {
}
STATIC MP_DEFINE_CONST_FUN_OBJ_1(mod_os_remove_obj, mod_os_remove);
+STATIC mp_obj_t mod_os_rename(mp_obj_t old_path_in, mp_obj_t new_path_in) {
+ const char *old_path = mp_obj_str_get_str(old_path_in);
+ const char *new_path = mp_obj_str_get_str(new_path_in);
+
+ int r = rename(old_path, new_path);
+
+ RAISE_ERRNO(r, errno);
+
+ return mp_const_none;
+}
+STATIC MP_DEFINE_CONST_FUN_OBJ_2(mod_os_rename_obj, mod_os_rename);
+
+STATIC mp_obj_t mod_os_rmdir(mp_obj_t path_in) {
+ const char *path = mp_obj_str_get_str(path_in);
+
+ int r = rmdir(path);
+
+ RAISE_ERRNO(r, errno);
+
+ return mp_const_none;
+}
+STATIC MP_DEFINE_CONST_FUN_OBJ_1(mod_os_rmdir_obj, mod_os_rmdir);
+
STATIC mp_obj_t mod_os_system(mp_obj_t cmd_in) {
const char *cmd = mp_obj_str_get_str(cmd_in);
@@ -236,6 +260,8 @@ STATIC const mp_rom_map_elem_t mp_module_os_globals_table[] = {
#endif
{ MP_ROM_QSTR(MP_QSTR_system), MP_ROM_PTR(&mod_os_system_obj) },
{ MP_ROM_QSTR(MP_QSTR_remove), MP_ROM_PTR(&mod_os_remove_obj) },
+ { MP_ROM_QSTR(MP_QSTR_rename), MP_ROM_PTR(&mod_os_rename_obj) },
+ { MP_ROM_QSTR(MP_QSTR_rmdir), MP_ROM_PTR(&mod_os_rmdir_obj) },
{ MP_ROM_QSTR(MP_QSTR_getenv), MP_ROM_PTR(&mod_os_getenv_obj) },
{ MP_ROM_QSTR(MP_QSTR_mkdir), MP_ROM_PTR(&mod_os_mkdir_obj) },
{ MP_ROM_QSTR(MP_QSTR_ilistdir), MP_ROM_PTR(&mod_os_ilistdir_obj) },