more snippet tests

RustPython · youknowone · Mar 25, 2025 · Mar 3, 2025 · Mar 3, 2025 · Mar 4, 2025
commit af7083c5b24e07bcda5367043ac22a3c1d90a774
diff --git a/extra_tests/snippets/builtins_ctypes.py b/extra_tests/snippets/builtins_ctypes.py
@@ -5,6 +5,7 @@
 from _ctypes import sizeof
 from _ctypes import _SimpleCData
 from _ctypes import CFuncPtr as _CFuncPtr
+import ctypes

 from struct import calcsize as _calcsize

@@ -263,12 +264,62 @@ def LoadLibrary(self, name):

 cdll = LibraryLoader(CDLL)

-test_byte_array = create_string_buffer(b"Hello, World!")
-assert test_byte_array._length_ == 14
+test_byte_array = create_string_buffer(b"Hello, World!\n")
+assert test_byte_array._length_ == 15

 if _os.name == "posix" or _sys.platform == "darwin":
    pass
 else:
+    import os
+
    libc = cdll.msvcrt
    libc.rand()
+    i = c_int(1)
+    print("start srand")
+    print(libc.srand(i))
+    print(test_byte_array)
+    print(test_byte_array._type_)
+    print("start printf")
    libc.printf(test_byte_array)
+
+    # windows pip support
+
+    def get_win_folder_via_ctypes(csidl_name: str) -> str:
+        """Get folder with ctypes."""
+        # There is no 'CSIDL_DOWNLOADS'.
+        # Use 'CSIDL_PROFILE' (40) and append the default folder 'Downloads' instead.
+        # https://learn.microsoft.com/en-us/windows/win32/shell/knownfolderid
+
+        import ctypes  # noqa: PLC0415
+
+        csidl_const = {
+            "CSIDL_APPDATA": 26,
+            "CSIDL_COMMON_APPDATA": 35,
+            "CSIDL_LOCAL_APPDATA": 28,
+            "CSIDL_PERSONAL": 5,
+            "CSIDL_MYPICTURES": 39,
+            "CSIDL_MYVIDEO": 14,
+            "CSIDL_MYMUSIC": 13,
+            "CSIDL_DOWNLOADS": 40,
+            "CSIDL_DESKTOPDIRECTORY": 16,
+        }.get(csidl_name)
+        if csidl_const is None:
+            msg = f"Unknown CSIDL name: {csidl_name}"
+            raise ValueError(msg)
+
+        buf = ctypes.create_unicode_buffer(1024)
+        windll = getattr(ctypes, "windll")  # noqa: B009 # using getattr to avoid false positive with mypy type checker
+        windll.shell32.SHGetFolderPathW(None, csidl_const, None, 0, buf)
+
+        # Downgrade to short path name if it has high-bit chars.
+        if any(ord(c) > 255 for c in buf):  # noqa: PLR2004
+            buf2 = ctypes.create_unicode_buffer(1024)
+            if windll.kernel32.GetShortPathNameW(buf.value, buf2, 1024):
+                buf = buf2
+
+        if csidl_name == "CSIDL_DOWNLOADS":
+            return os.path.join(buf.value, "Downloads")  # noqa: PTH118
+
+        return buf.value
+
+    print(get_win_folder_via_ctypes("CSIDL_DOWNLOADS"))