Skip to content
This repository has been archived by the owner on Feb 21, 2025. It is now read-only.

Commit

Permalink
Add PyBytes_FromStringAndSize API wrapper (#2)
Browse files Browse the repository at this point in the history
  • Loading branch information
rw-access authored Aug 18, 2022
1 parent 7802429 commit fb53c94
Show file tree
Hide file tree
Showing 2 changed files with 25 additions and 10 deletions.
28 changes: 18 additions & 10 deletions bytes.go
Original file line number Diff line number Diff line change
Expand Up @@ -14,50 +14,58 @@ package python3
import "C"
import "unsafe"

//Bytes : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_Type
// Bytes : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_Type
var Bytes = togo((*C.PyObject)(unsafe.Pointer(&C.PyBytes_Type)))

//PyBytes_Check : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_Check
// PyBytes_Check : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_Check
func PyBytes_Check(o *PyObject) bool {
return C._go_PyBytes_Check(toc(o)) != 0
}

//PyBytes_CheckExact : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_CheckExact
// PyBytes_CheckExact : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_CheckExact
func PyBytes_CheckExact(o *PyObject) bool {
return C._go_PyBytes_CheckExact(toc(o)) != 0
}

//PyBytes_FromString : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_FromString
// PyBytes_FromString : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_FromString
func PyBytes_FromString(str string) *PyObject {
cstr := C.CString(str)
defer C.free(unsafe.Pointer(cstr))

return togo(C.PyBytes_FromString(cstr))
}

//PyBytes_FromObject : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_FromObject
// PyBytes_FromStringAndSize : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_AsString
func PyBytes_FromStringAndSize(str string) *PyObject {
cstr := C.CString(str)
defer C.free(unsafe.Pointer(cstr))

return togo(C.PyBytes_FromStringAndSize(cstr, C.Py_ssize_t(len(str))))
}

// PyBytes_FromObject : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_FromObject
func PyBytes_FromObject(o *PyObject) *PyObject {
return togo(C.PyBytes_FromObject(toc(o)))
}

//PyBytes_Size : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_Size
// PyBytes_Size : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_Size
func PyBytes_Size(o *PyObject) int {
return int(C.PyBytes_Size(toc(o)))
}

//PyBytes_AsString : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_AsString
// PyBytes_AsString : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_AsString
func PyBytes_AsString(o *PyObject) string {
return C.GoStringN(C.PyBytes_AsString(toc(o)), C.int(C.PyBytes_Size(toc(o))))
return C.GoStringN(C.PyBytes_AsString(toc(o)), C.int(C.PyBytes_Size(toc(o))))
}

//PyBytes_Concat : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_Concat
// PyBytes_Concat : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_Concat
func PyBytes_Concat(bytes, newpart *PyObject) *PyObject {
cbytes := toc(bytes)
C.PyBytes_Concat(&cbytes, toc(newpart))
return togo(cbytes)
}

//PyBytes_ConcatAndDel : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_ConcatAndDel
// PyBytes_ConcatAndDel : https://docs.python.org/3/c-api/bytes.html#c.PyBytes_ConcatAndDel
func PyBytes_ConcatAndDel(bytes, newpart *PyObject) *PyObject {
cbytes := toc(bytes)
C.PyBytes_ConcatAndDel(&cbytes, toc(newpart))
Expand Down
7 changes: 7 additions & 0 deletions import_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -124,6 +124,13 @@ func TestExecCodeModule(t *testing.T) {
module := PyImport_ExecCodeModule("test_module", code)
assert.NotNil(t, module)

testModuleStr := PyUnicode_FromString("test_module")
defer testModuleStr.DecRef()

pyModule := PyImport_GetModule(testModuleStr)
defer pyModule.DecRef()

assert.NotNil(t, pyModule)
}

func TestExecCodeModuleEx(t *testing.T) {
Expand Down

0 comments on commit fb53c94

Please sign in to comment.