@@ -420,18 +420,6 @@ fn get_addr_tuple(vm: &mut VirtualMachine, addr: SocketAddr) -> PyResult {
420420}
421421
422422pub fn mk_module ( ctx : & PyContext ) ->PyObjectRef {
423- let py_mod = ctx. new_module ( & "socket" . to_string ( ) , ctx. new_scope ( None ) ) ;
424-
425- ctx. set_attr ( & py_mod, "AF_INET" , ctx. new_int ( AddressFamily :: Inet as i32 ) ) ;
426-
427- ctx. set_attr (
428- & py_mod,
429- "SOCK_STREAM" ,
430- ctx. new_int ( SocketKind :: Stream as i32 ) ,
431- ) ;
432-
433- ctx. set_attr ( & py_mod, "SOCK_DGRAM" , ctx. new_int ( SocketKind :: Dgram as i32 ) ) ;
434-
435423let socket =py_class ! ( ctx, "socket" , ctx. object( ) , {
436424"__new__" => ctx. new_rustfunc( socket_new) ,
437425"connect" => ctx. new_rustfunc( socket_connect) ,
@@ -445,7 +433,11 @@ pub fn mk_module(ctx: &PyContext) -> PyObjectRef {
445433"sendto" => ctx. new_rustfunc( socket_sendto) ,
446434"recvfrom" => ctx. new_rustfunc( socket_recvfrom) ,
447435} ) ;
448- ctx. set_attr ( & py_mod, "socket" , socket. clone ( ) ) ;
449436
450- py_mod
437+ py_module ! ( ctx, "socket" , {
438+ "AF_INET" => ctx. new_int( AddressFamily :: Inet as i32 ) ,
439+ "SOCK_STREAM" => ctx. new_int( SocketKind :: Stream as i32 ) ,
440+ "SOCK_DGRAM" => ctx. new_int( SocketKind :: Dgram as i32 ) ,
441+ "socket" => socket. clone( ) ,
442+ } )
451443}