phi/test/aux_async_model.py

160 lines
4.3 KiB
Python
Raw Normal View History

2020-12-27 19:48:02 +01:00
# -*- coding: utf-8 -*-
2019-04-28 15:34:37 +02:00
import asyncio
2020-08-31 11:02:03 +02:00
from pprint import pprint as pp
2019-04-28 15:34:37 +02:00
2020-11-11 22:25:02 +01:00
from phi.async_ldap.new_model import (
Hackers,
User,
2019-04-28 15:34:37 +02:00
Robots,
Service,
2020-08-31 11:02:03 +02:00
Group,
2020-11-11 22:25:02 +01:00
Congregations,
2019-04-28 15:34:37 +02:00
)
2020-11-11 22:25:02 +01:00
from phi.async_ldap.mixins import build_heritage
2020-08-31 11:02:03 +02:00
from phi.async_ldap.client import AsyncClient
2020-11-11 22:25:02 +01:00
import phi.exceptions as e
2019-04-28 15:34:37 +02:00
2020-08-31 11:02:03 +02:00
async def dlv(h, cls):
return [el async for el in build_heritage(h, cls)]
2019-04-28 15:34:37 +02:00
cl = AsyncClient(
"ldap://localhost",
port=389,
encryption=True,
2020-08-31 11:02:03 +02:00
# validate=True,
ca_cert="/home/leo/Documents/coding/phi/openldap/cert.pem",
2019-04-28 15:34:37 +02:00
username="root",
password="root",
base_dn="dc=unit,dc=macaomilano,dc=org",
attribute_id="cn",
)
2020-08-31 11:02:03 +02:00
async def get_all_children():
h = Hackers(cl)
r = Robots(cl)
g = Congregations(cl)
hackers = await dlv(h, User)
robots = await dlv(r, Service)
groups = await dlv(g, Group)
return (hackers, robots, groups)
2020-11-11 22:25:02 +01:00
async def get_members(group):
return [el async for el in group]
async def print_async(label, awaitable):
print(label)
2020-08-31 11:02:03 +02:00
result = await awaitable
pp(result)
async def describe(obj):
2020-11-11 22:25:02 +01:00
return await obj.describe()
async def _await(awaitable):
return await awaitable
def sync_await(awaitable):
return asyncio.run(_await(awaitable))
h = Hackers(cl)
r = Robots(cl)
c = Congregations(cl)
# asyncio.run(print_async("hackers:", describe(h)))
# asyncio.run(print_async("conte_mascetti:", describe(User(cl, "conte_mascetti"))))
# asyncio.run(print_async("robots:", describe(r)))
# asyncio.run(print_async("phi:", describe(Service(cl, "phi"))))
# asyncio.run(print_async("congregations:", describe(c)))
# asyncio.run(print_async("GitUsers:", describe(Group(cl, "GitUsers"))))
# asyncio.run(print_async("Hackers members:", get_members(h)))
# asyncio.run(print_async("Robots members:", get_members(r)))
# asyncio.run(print_async("Congregations members:", get_members(c)))
2020-08-31 11:02:03 +02:00
2020-11-11 22:25:02 +01:00
#
async def add_new(obj, name, **kw):
try:
_new = obj(cl, name, **kw)
await _new.save()
except e.PhiEntryExists as err:
print(f"Failed add: {repr(err)}")
2020-08-31 11:02:03 +02:00
2020-11-11 22:25:02 +01:00
async def safe_search(group, name):
try:
res = await group.search(name)
print("Search result:", res)
return res
except e.PhiEntryDoesNotExist as err:
print(f"Failed search: {repr(err)}")
async def safe_delete(obj, cascade=None):
try:
if cascade:
obj.delete_cascade = cascade
await obj.delete()
except Exception as err:
print(f"Failed delete: {repr(err)}")
async def add_member(group, member):
await group.add_member(member)
async def remove_member(group, member):
await group.remove_member(member)
# asyncio.run(safe_search(h, "pippo"))
# asyncio.run(
# add_new(User, "pippo", cn="Pippo (Goofy)", sn="Pippo", mail="pippo@unit.info")
# )
# asyncio.run(safe_search(h, "pippo"))
# asyncio.run(
# add_new(User, "pippo", cn="Pippo (Goofy)", sn="Pippo", mail="pippo@unit.net")
# )
# asyncio.run(safe_delete(asyncio.run(safe_search(h, "pippo"))))
# asyncio.run(print_async("Hackers members:", get_members(h)))
# asyncio.run(safe_delete(h))
# asyncio.run(print_async("Hackers members:", get_members(h)))
# asyncio.run(safe_delete(h, True))
# asyncio.run(print_async("Hackers members:", get_members(h)))
# asyncio.run(safe_search(r, "phi"))
# asyncio.run(print_async("Robots members:", get_members(r)))
# asyncio.run(add_new(Service, "db", userPassword="lolpassword"))
# asyncio.run(print_async("Robots members:", get_members(r)))
asyncio.run(safe_search(c, "GitUsers"))
asyncio.run(print_async("Congregations members:", get_members(c)))
asyncio.run(
add_new(Group, "naughty", member=[User(cl, "conte_mascetti"), User(cl, "necchi")])
)
asyncio.run(print_async("Congregations members:", get_members(c)))
asyncio.run(safe_delete(Group(cl, "naughty")))
asyncio.run(print_async("Congregations members:", get_members(c)))
asyncio.run(
add_new(Group, "naughty", member=[User(cl, "conte_mascetti"), User(cl, "necchi")])
)
asyncio.run(print_async("Congregations members:", get_members(c)))
print("==> HERE <==")
naughty = sync_await(Group(cl, "naughty").sync())
print("NAUGHTY =>>", [m for m in naughty.get_members()])
asyncio.run(add_member(naughty, User(cl, "perozzi")))
print("NAUGHTY =>>", [m for m in naughty.get_members()])
asyncio.run(remove_member(naughty, User(cl, "conte_mascetti")))
print("NAUGHTY =>>", [m for m in naughty.get_members()])