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

add conio module

Browse Source
This commit is contained in:
blueloveTH 2024-10-01 14:00:31 +08:00
parent b6caac01de
commit f65eec9add
4 changed files with 34 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
include/pocketpy/interpreter/modules.h
View File

@ -1,6 +1,7 @@
#pragma once
void pk__add_module_pkpy();
void pk__add_module_conio();
void pk__add_module_os();
void pk__add_module_sys();
void pk__add_module_math();

2
include/typings/conio.pyi Normal file
View File

@ -0,0 +1,2 @@
def _kbhit() -> int: ...
def _getch() -> int: ...

2
src/interpreter/vm.c
View File

@ -205,6 +205,8 @@ void VM__ctor(VM* self) {
// add modules
pk__add_module_pkpy();
pk__add_module_conio();
pk__add_module_os();
pk__add_module_sys();
pk__add_module_math();

31
src/modules/pkpy.c
View File

@ -1,5 +1,32 @@
#include "pocketpy/pocketpy.h"
void pk__add_module_pkpy() {
py_newmodule("pkpy");
void pk__add_module_pkpy() { py_newmodule("pkpy"); }
#ifdef _WIN32
#include <conio.h>
static bool conio__kbhit(int argc, py_Ref argv) {
PY_CHECK_ARGC(0);
int ret = _kbhit();
py_newint(py_retval(), ret);
return true;
}
static bool conio__getch(int argc, py_Ref argv) {
PY_CHECK_ARGC(0);
int ret = _getch();
py_newint(py_retval(), ret);
return true;
}
#endif
void pk__add_module_conio() {
py_Ref mod = py_newmodule("conio");
#ifdef _WIN32
py_bindfunc(mod, "_kbhit", conio__kbhit);
py_bindfunc(mod, "_getch", conio__getch);
#endif
}