U
|f1 @ s, d Z ddlZddlZddlZddlZddlZddlmZ ddlZddlm Z ddlm
Z
ddlm Z ddl m
Z
ddl mZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlm!Z! ddlm"Z" dd lm#Z# dd lm$Z$ dd lm%Z% erd d Z!d!d" Z&G d#d$ d$eZ'e"e(ed%e(eoe
d&G d'd( d(e'Z)e(ed)G d*d+ d+e)Z*G d,d- d-e'Z+e,d.kr(dd/l-m.Z. e.e/ dS )0a> Notes about unicode handling in psutil
======================================.
Starting from version 5.3.0 psutil adds unicode support, see:
https://github.com/giampaolo/psutil/issues/1040
The notes below apply to *any* API returning a string such as
process exe(), cwd() or username():
* all strings are encoded by using the OS filesystem encoding
(sys.getfilesystemencoding()) which varies depending on the platform
(e.g. "UTF-8" on macOS, "mbcs" on Win)
* no API call is supposed to crash with UnicodeDecodeError
* instead, in case of badly encoded data returned by the OS, the
following error handlers are used to replace the corrupted characters in
the string:
* Python 3: sys.getfilesystemencodeerrors() (PY 3.6+) or
"surrogatescape" on POSIX and "replace" on Windows
* Python 2: "replace"
* on Python 2 all APIs return bytes (str type), never unicode
* on Python 2, you can go back to unicode by doing:
>>> unicode(p.exe(), sys.getdefaultencoding(), errors="replace")
For a detailed explanation of how psutil handles unicode see #1040.
Tests
=====
List of APIs returning or dealing with a string:
('not tested' means they are not tested to deal with non-ASCII strings):
* Process.cmdline()
* Process.cwd()
* Process.environ()
* Process.exe()
* Process.memory_maps()
* Process.name()
* Process.net_connections('unix')
* Process.open_files()
* Process.username() (not tested)
* disk_io_counters() (not tested)
* disk_partitions() (not tested)
* disk_usage(str)
* net_connections('unix')
* net_if_addrs() (not tested)
* net_if_stats() (not tested)
* net_io_counters() (not tested)
* sensors_fans() (not tested)
* sensors_temperatures() (not tested)
* users() (not tested)
* WindowsService.binpath() (not tested)
* WindowsService.description() (not tested)
* WindowsService.display_name() (not tested)
* WindowsService.name() (not tested)
* WindowsService.status() (not tested)
* WindowsService.username() (not tested)
In here we create a unicode path with a funky non-ASCII name and (where
possible) make psutil return it back (e.g. on name(), exe(), open_files(),
etc.) and make sure that:
* psutil never crashes with UnicodeDecodeError
* the returned path matches
N)closing)BSD)POSIX)WINDOWS)PY3)super)APPVEYOR)ASCII_FS)
CI_TESTING) HAS_ENVIRON)HAS_MEMORY_MAPS)HAS_NET_CONNECTIONS_UNIX)INVALID_UNICODE_SUFFIX)PYPY
TESTFN_PREFIX)UNICODE_SUFFIX)PsutilTestCase)bind_unix_socket)chdir)copyload_shared_lib)
create_py_exe)
get_testfn)
safe_mkdir safe_rmpath) serialrun)skip_on_access_denied)spawn_testproc) terminatec C s8 ddl m} z
|| W S tk
r2 t Y nX d S )Nr r ) psutil.testsr Z WindowsError traceback print_exc)pathZrm r$ L/opt/alt/python38/lib64/python3.8/site-packages/psutil/tests/test_unicode.pyr r s
r c C s d}t | d}zhz.find_sockr& rk rl )kind)r r3 r r+ r r8 r9 r rN rm rR rn rE rS )r; ru rX ro rt rp r$ r$ r% test_net_connections s
z TestFSAPIs.test_net_connectionsc C s, | j d }| t| t| t| d S r^ )r2 r` r r rN
disk_usage)r; rb r$ r$ r% test_disk_usage2 s
zTestFSAPIs.test_disk_usagerk z&ctypes does not support unicode on PY2zunstable on PYPYc sr t | jd\}dd fddt D }dd |D }| || |D ]}| |t qRW 5 Q R X d S )Nr& c S s t jt j| S )N)rI r# realpathrT )rV r$ r$ r% normpath@ s z-TestFSAPIs.test_memory_maps..normpathc s g | ]} |j qS r$ )r# .0xr{ r$ r%
C s z/TestFSAPIs.test_memory_maps..c S s g | ]}t |kr|qS r$ r r| r$ r$ r% r G s )r r3 rN rO Z memory_mapsZassertInrR rE )r; Z
funky_pathZlibpathsr# r$ r r% test_memory_maps8 s
zTestFSAPIs.test_memory_mapsN)r< r= r> __doc__r r3 rL rW rZ r] rc r8 skipIfr r rj r rq r
r rw ry r r r r$ r$ r$ r% rA s(
rA zunreliable on CIc @ s e Zd ZdZeZdd ZdS )TestFSAPIsWithInvalidPathz-Test FS APIs with a funky, invalid path name.c C s dS )NTr$ r: r$ r$ r% rL S s z1TestFSAPIsWithInvalidPath.expect_exact_path_matchN)r< r= r> r r r3 rL r$ r$ r$ r% r M s r c @ sB e Zd ZdZerendZee dee
o.e ddd Z dS )
TestNonFSAPISz&Unicode tests for non fs-related APIs. èrk zsegfaults on PYPY + WINDOWSc C st t j }| j|d< | j|d}t|j}| }| D ] \}}| |t
| |t
q<| |d | j d S )NZ FUNNY_ARG)env) rI environcopyr3 r rN rO rP itemsrR rE rS )r; r r- rV kvr$ r$ r% test_proc_environb s
z TestNonFSAPIS.test_proc_environN)
r< r= r> r r r r3 r8 r r r r r r$ r$ r$ r% r ] s
r __main__)
run_from_name)0r rI r) r! r8 rF
contextlibr rN r r r Zpsutil._compatr r r r r r
r r r
r r r r r r r r r r r r r r r r r. r/ r rA r r r< Zpsutil.tests.runnerr __file__r$ r$ r$ r% sb C