Skip to content
Snippets Groups Projects
Verified Commit f8a8f0d7 authored by rahix's avatar rahix
Browse files

feat(os): Make os.listdir() callable without args


For convenience, allow calling os.listdir() without any argument.
This is equivalent to calling os.listdir("").

Signed-off-by: default avatarRahix <rahix@rahix.de>
parent fba3afaf
No related branches found
No related tags found
No related merge requests found
......@@ -91,10 +91,16 @@ static mp_obj_t mp_os_reset(void)
}
static MP_DEFINE_CONST_FUN_OBJ_0(reset_obj, mp_os_reset);
static mp_obj_t mp_os_listdir(mp_obj_t py_path)
static mp_obj_t mp_os_listdir(size_t n_args, const mp_obj_t *args)
{
const char *path = mp_obj_str_get_str(py_path);
int fd = epic_file_opendir(path);
const char *path;
if (n_args == 1) {
path = mp_obj_str_get_str(args[0]);
} else {
path = "";
}
int fd = epic_file_opendir(path);
if (fd < 0) {
mp_raise_OSError(-fd);
......@@ -118,7 +124,7 @@ static mp_obj_t mp_os_listdir(mp_obj_t py_path)
epic_file_close(fd);
return MP_OBJ_FROM_PTR(list);
}
static MP_DEFINE_CONST_FUN_OBJ_1(listdir_obj, mp_os_listdir);
static MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(listdir_obj, 0, 1, mp_os_listdir);
static mp_obj_t mp_os_unlink(mp_obj_t py_path)
{
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment