|
28 | 28 | #include "py/mpconfig.h" |
29 | 29 | #if MICROPY_VFS_FAT |
30 | 30 |
|
| 31 | +#include <string.h> |
31 | 32 | #include "py/nlr.h" |
32 | 33 | #include "py/runtime.h" |
33 | 34 | #include "lib/fatfs/ff.h" |
@@ -136,12 +137,26 @@ STATIC mp_obj_t fat_vfs_chdir(mp_obj_t path_in) { |
136 | 137 | } |
137 | 138 | STATIC MP_DEFINE_CONST_FUN_OBJ_1(fat_vfs_chdir_obj, fat_vfs_chdir); |
138 | 139 |
|
| 140 | +/// Get the current directory. |
| 141 | +STATIC mp_obj_t fat_vfs_getcwd(void) { |
| 142 | + char buf[MICROPY_ALLOC_PATH_MAX + 1]; |
| 143 | + FRESULT res = f_getcwd(buf, sizeof buf); |
| 144 | + |
| 145 | + if (res != FR_OK) { |
| 146 | + nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_OSError, MP_OBJ_NEW_SMALL_INT(fresult_to_errno_table[res]))); |
| 147 | + } |
| 148 | + |
| 149 | + return mp_obj_new_str(buf, strlen(buf), false); |
| 150 | +} |
| 151 | +STATIC MP_DEFINE_CONST_FUN_OBJ_0(fat_vfs_getcwd_obj, fat_vfs_getcwd); |
| 152 | + |
139 | 153 | STATIC const mp_rom_map_elem_t fat_vfs_locals_dict_table[] = { |
140 | 154 | { MP_ROM_QSTR(MP_QSTR_mkfs), MP_ROM_PTR(&fat_vfs_mkfs_obj) }, |
141 | 155 | { MP_ROM_QSTR(MP_QSTR_open), MP_ROM_PTR(&fat_vfs_open_obj) }, |
142 | 156 | { MP_ROM_QSTR(MP_QSTR_listdir), MP_ROM_PTR(&fat_vfs_listdir_obj) }, |
143 | 157 | { MP_ROM_QSTR(MP_QSTR_mkdir), MP_ROM_PTR(&fat_vfs_mkdir_obj) }, |
144 | 158 | { MP_ROM_QSTR(MP_QSTR_chdir), MP_ROM_PTR(&fat_vfs_chdir_obj) }, |
| 159 | + { MP_ROM_QSTR(MP_QSTR_getcwd), MP_ROM_PTR(&fat_vfs_getcwd_obj) }, |
145 | 160 | { MP_ROM_QSTR(MP_QSTR_remove), MP_ROM_PTR(&fat_vfs_remove_obj) }, |
146 | 161 | { MP_ROM_QSTR(MP_QSTR_rename), MP_ROM_PTR(&fat_vfs_rename_obj) }, |
147 | 162 | }; |
|
0 commit comments