mirror of
https://github.com/pocketpy/pocketpy
synced 2025-10-20 11:30:18 +00:00
109 lines
4.2 KiB
C++
109 lines
4.2 KiB
C++
|
|
/**
|
|
* This example illustrate use of Dict in PocketPy.
|
|
* It creates a python module named "employee" and bind four functions to it.
|
|
* It exercises setting and getting elements in a Dict.
|
|
*/
|
|
|
|
|
|
#include "pocketpy.h"
|
|
|
|
using namespace pkpy;
|
|
|
|
|
|
int main(){
|
|
// Create a virtual machine
|
|
VM* vm = new VM();
|
|
|
|
// Create "employee" module
|
|
PyObject* employee_module = vm->new_module("employee");
|
|
|
|
// Bind a function named "get_first_name" to the module
|
|
vm->bind(employee_module, "get_first_name(employee: Dict) -> str",
|
|
"Returns first_name of the employee", // docstring
|
|
[](VM* vm, ArgsView args){
|
|
// Cast the argument to a dictionary
|
|
Dict& employee = CAST(Dict&, args[0]);
|
|
// Access the first_name field and return it
|
|
return employee.try_get(VAR("first_name"));
|
|
});
|
|
|
|
// Bind a function named "get_last_name" to the module
|
|
vm->bind(employee_module, "get_last_name(employee: Dict) -> str",
|
|
"Returns last_name of the employee", // docstring
|
|
[](VM* vm, ArgsView args){
|
|
// Cast the argument to a dictionary
|
|
Dict& employee = CAST(Dict&, args[0]);
|
|
// Access the last_name field and return it
|
|
return employee.try_get(VAR("last_name"));
|
|
});
|
|
|
|
// Bind a function named "get_salary" to the module
|
|
// It accepts a dictionary as argument and returns a float
|
|
vm->bind(employee_module, "get_salary(employee: Dict) -> float",
|
|
"Returns salary of the employee", // docstring
|
|
[](VM* vm, ArgsView args){
|
|
// Cast the argument to a dictionary
|
|
Dict& employee = CAST(Dict&, args[0]);
|
|
// Access the salary field and return it
|
|
return employee.try_get(VAR("salary"));
|
|
});
|
|
|
|
// Bind a function named "love_coding" to the module
|
|
// It accepts a dictionary as argument and returns a bool
|
|
vm->bind(employee_module, "love_coding(employee: Dict) -> bool",
|
|
"Returns Yes if the employee loves coding, No otherwise", // docstring
|
|
[](VM* vm, ArgsView args){
|
|
// Cast the argument to a dictionary
|
|
Dict& employee = CAST(Dict&, args[0]);
|
|
|
|
// Access the hobbies field and cast it to a list
|
|
List& hobbies = CAST(List&, employee.try_get(VAR("hobbies")));
|
|
|
|
// Iterate through the list and check if the employee loves coding
|
|
for(auto& e : hobbies){
|
|
if(CAST(Str&, e) == Str("Coding")){
|
|
return VAR("Yes");
|
|
}
|
|
}
|
|
return VAR("No");
|
|
});
|
|
|
|
// Create employee dictionary covert it to a python object
|
|
Dict employee(vm);
|
|
employee.set(VAR("first_name"), VAR("John"));
|
|
employee.set(VAR("last_name"), VAR("Doe"));
|
|
employee.set(VAR("age"), VAR(30));
|
|
employee.set(VAR("salary"), VAR(10000.0));
|
|
List hobbies;
|
|
hobbies.push_back(VAR("Reading"));
|
|
hobbies.push_back(VAR("Walking"));
|
|
hobbies.push_back(VAR("Coding"));
|
|
employee.set(VAR("hobbies"), VAR(std::move(hobbies)));
|
|
PyObject* employee_obj = VAR(std::move(employee));
|
|
|
|
// Call the "get_first_name" function
|
|
PyObject* f_get_first_name = employee_module->attr("get_first_name");
|
|
PyObject* first_name = vm->call(f_get_first_name, employee_obj);
|
|
std::cout << "First name: " << CAST(Str&, first_name) << std::endl; // First name: John
|
|
|
|
// Call the "get_last_name" function
|
|
PyObject* f_get_last_name = employee_module->attr("get_last_name");
|
|
PyObject* last_name = vm->call(f_get_last_name, employee_obj);
|
|
std::cout << "Last name: " << CAST(Str&, last_name) << std::endl; // Last name: Doe
|
|
|
|
// Call the "get_salary" function
|
|
PyObject* f_get_salary = employee_module->attr("get_salary");
|
|
PyObject* salary = vm->call(f_get_salary, employee_obj);
|
|
std::cout << "Salary: "<< CAST(double, salary) << std::endl; // Salary: 10000
|
|
|
|
// Call the "love_coding" function
|
|
PyObject* f_love_coding = employee_module->attr("love_coding");
|
|
PyObject* love_coding = vm->call(f_love_coding, employee_obj);
|
|
std::cout << "Loves coding: " << CAST(Str&, love_coding) << std::endl; // Loves coding: Yes
|
|
|
|
// Dispose the virtual machine
|
|
delete vm;
|
|
return 0;
|
|
}
|