Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-10-20 19:40:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

up

Browse Source
This commit is contained in:
blueloveTH 2023-03-27 13:35:59 +08:00
parent 7c6ef3caba
commit dc72796e2c
3 changed files with 24 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/io.h
View File

@ -88,10 +88,17 @@ void add_module_io(VM* vm){
void add_module_os(VM* vm){
PyVar mod = vm->new_module("os");
// Working directory is shared by all VMs!!
vm->bind_func<0>(mod, "getcwd", [](VM* vm, const Args& args){
return VAR(std::filesystem::current_path().string());
});
vm->bind_func<1>(mod, "chdir", [](VM* vm, const Args& args){
std::filesystem::path path(CAST(Str&, args[0]).c_str());
std::filesystem::current_path(path);
return vm->None;
});
vm->bind_func<1>(mod, "listdir", [](VM* vm, const Args& args){
std::filesystem::path path(CAST(Str&, args[0]).c_str());
std::filesystem::directory_iterator di;

23
src/main.cpp
View File

@ -1,6 +1,7 @@
#include <fstream>
#include "pocketpy.h"
#include <filesystem>
#include "pocketpy.h"
#ifdef _WIN32
@ -58,17 +59,23 @@ int main(int argc, char** argv){
}
if(argc == 2){
std::string filename = argv[1];
if(filename == "-h" || filename == "--help") goto __HELP;
std::string argv_1 = argv[1];
if(argv_1 == "-h" || argv_1 == "--help") goto __HELP;
std::ifstream file(filename);
if(!file.is_open()){
std::cerr << "File not found: " << filename << std::endl;
std::filesystem::path filepath(argv[1]);
if(!std::filesystem::exists(filepath)){
std::cerr << "File not found: " << argv_1 << std::endl;
return 1;
}
}
std::ifstream file(filepath);
if(!file.is_open()) return 1;
std::string src((std::istreambuf_iterator<char>(file)), std::istreambuf_iterator<char>());
// set parent path as cwd
std::filesystem::current_path(filepath.parent_path());
pkpy::PyVarOrNull ret = nullptr;
ret = vm->exec(src.c_str(), filename, pkpy::EXEC_MODE);
ret = vm->exec(src.c_str(), argv_1, pkpy::EXEC_MODE);
pkpy_delete(vm);
return ret != nullptr ? 0 : 1;
}

2
tests/70_file.py
View File

@ -18,4 +18,6 @@ with open('123.txt', 'a') as f:
with open('123.txt', 'r') as f:
assert f.read() == '123456' + '测试'
assert os.path_exists('123.txt')
os.remove('123.txt')
assert not os.path_exists('123.txt')