mirror of
https://github.com/HuFlungDu/pylibmeshctrl.git
synced 2026-02-20 13:42:11 +00:00
Added Session to the __init__ file, and changed docs and test accordingly
This commit is contained in:
@@ -44,8 +44,7 @@ Usage
|
|||||||
-----
|
-----
|
||||||
|
|
||||||
This module is implemented as a primarily asynchronous library
|
This module is implemented as a primarily asynchronous library
|
||||||
(asyncio), mostly through the ``Session`` class, which is exported as
|
(asyncio), mostly through the `Session <https://pylibmeshctrl.readthedocs.io/en/latest/api/meshctrl.html#meshctrl.session.Session>`__ class. Because the library is asynchronous, you must wait for it to be
|
||||||
default. Because the library is asynchronous, you must wait for it to be
|
|
||||||
initialized before interacting with the server. The preferred way to do
|
initialized before interacting with the server. The preferred way to do
|
||||||
this is to use the async context manager pattern:
|
this is to use the async context manager pattern:
|
||||||
|
|
||||||
@@ -53,12 +52,12 @@ this is to use the async context manager pattern:
|
|||||||
|
|
||||||
import meshctrl
|
import meshctrl
|
||||||
|
|
||||||
async with meshctrl.session.Session(url, **options):
|
async with meshctrl.Session(url, **options):
|
||||||
print(await session.list_users())
|
print(await session.list_users())
|
||||||
...
|
...
|
||||||
|
|
||||||
However, if you prefer to instantiate the object yourself, you can
|
However, if you prefer to instantiate the object yourself, you can
|
||||||
simply use the ``initialized`` property:
|
simply use the `initialized <https://pylibmeshctrl.readthedocs.io/en/latest/api/meshctrl.html#meshctrl.session.Session.initialized>`__ property:
|
||||||
|
|
||||||
.. code:: python
|
.. code:: python
|
||||||
|
|
||||||
@@ -66,7 +65,7 @@ simply use the ``initialized`` property:
|
|||||||
await session.initialized.wait()
|
await session.initialized.wait()
|
||||||
|
|
||||||
Note that, in this case, you will be rquired to clean up tho session
|
Note that, in this case, you will be rquired to clean up tho session
|
||||||
using its ``close`` method.
|
using its `close <https://pylibmeshctrl.readthedocs.io/en/latest/api/meshctrl.html#meshctrl.session.Session.close>`__ method.
|
||||||
|
|
||||||
Session Parameters
|
Session Parameters
|
||||||
------------------
|
------------------
|
||||||
|
|||||||
@@ -22,9 +22,6 @@ class FileTransferError(MeshCtrlError):
|
|||||||
|
|
||||||
Attributes:
|
Attributes:
|
||||||
stats (dict): {"result" (str): Human readable result, "size" (int): number of bytes successfully transferred}
|
stats (dict): {"result" (str): Human readable result, "size" (int): number of bytes successfully transferred}
|
||||||
initialized (asyncio.Event): Event marking if the Session initialization has finished. Wait on this to wait for a connection.
|
|
||||||
alive (bool): Whether the session connection is currently alive
|
|
||||||
closed (asyncio.Event): Event that occurs when the session closes permanently
|
|
||||||
"""
|
"""
|
||||||
def __init__(self, message, stats):
|
def __init__(self, message, stats):
|
||||||
self.stats = stats
|
self.stats = stats
|
||||||
|
|||||||
@@ -1 +1 @@
|
|||||||
{"admin": "3U6zP4iIes5ISH15XxjYLjJcCdw9jU0m", "privileged": "aiIO0zLMGsU7++FYVDNxhlpYlZ1andRB", "unprivileged": "Cz9OMV1wkVd9pXdWi4lkBAAu6TMt43MA"}
|
{"admin": "vt9BbctCg59vcuxKPh8v2tbDjudjwyeX", "privileged": "BWl1vhSe0j0lBfoAkx1JLXLBOwIWc0st", "unprivileged": "vCuatfTQGq8bL2pxdvrNzF+Dc4xBq+5Z"}
|
||||||
@@ -7,7 +7,7 @@ import io
|
|||||||
import random
|
import random
|
||||||
|
|
||||||
async def test_commands(env):
|
async def test_commands(env):
|
||||||
async with meshctrl.session.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session:
|
async with meshctrl.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session:
|
||||||
mesh = await admin_session.add_device_group("test", description="This is a test group", amtonly=False, features=0, consent=0, timeout=10)
|
mesh = await admin_session.add_device_group("test", description="This is a test group", amtonly=False, features=0, consent=0, timeout=10)
|
||||||
try:
|
try:
|
||||||
with env.create_agent(mesh.short_meshid) as agent:
|
with env.create_agent(mesh.short_meshid) as agent:
|
||||||
@@ -53,7 +53,7 @@ async def test_commands(env):
|
|||||||
assert (await admin_session.remove_device_group(mesh.meshid, timeout=10)), "Failed to remove device group"
|
assert (await admin_session.remove_device_group(mesh.meshid, timeout=10)), "Failed to remove device group"
|
||||||
|
|
||||||
async def test_upload_download(env):
|
async def test_upload_download(env):
|
||||||
async with meshctrl.session.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session:
|
async with meshctrl.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session:
|
||||||
mesh = await admin_session.add_device_group("test", description="This is a test group", amtonly=False, features=0, consent=0, timeout=10)
|
mesh = await admin_session.add_device_group("test", description="This is a test group", amtonly=False, features=0, consent=0, timeout=10)
|
||||||
try:
|
try:
|
||||||
with env.create_agent(mesh.short_meshid) as agent:
|
with env.create_agent(mesh.short_meshid) as agent:
|
||||||
|
|||||||
@@ -8,14 +8,14 @@ import ssl
|
|||||||
import requests
|
import requests
|
||||||
|
|
||||||
async def test_sanity(env):
|
async def test_sanity(env):
|
||||||
async with meshctrl.session.Session(env.mcurl, user="unprivileged", password=env.users["unprivileged"], ignore_ssl=True) as s:
|
async with meshctrl.Session(env.mcurl, user="unprivileged", password=env.users["unprivileged"], ignore_ssl=True) as s:
|
||||||
print("\ninfo user_info: {}\n".format(await s.user_info()))
|
print("\ninfo user_info: {}\n".format(await s.user_info()))
|
||||||
print("\ninfo server_info: {}\n".format(await s.server_info()))
|
print("\ninfo server_info: {}\n".format(await s.server_info()))
|
||||||
pass
|
pass
|
||||||
|
|
||||||
async def test_ssl(env):
|
async def test_ssl(env):
|
||||||
try:
|
try:
|
||||||
async with meshctrl.session.Session(env.mcurl, user="unprivileged", password=env.users["unprivileged"], ignore_ssl=False) as s:
|
async with meshctrl.Session(env.mcurl, user="unprivileged", password=env.users["unprivileged"], ignore_ssl=False) as s:
|
||||||
pass
|
pass
|
||||||
except* ssl.SSLCertVerificationError:
|
except* ssl.SSLCertVerificationError:
|
||||||
pass
|
pass
|
||||||
|
|||||||
@@ -8,8 +8,8 @@ import io
|
|||||||
thisdir = os.path.dirname(os.path.realpath(__file__))
|
thisdir = os.path.dirname(os.path.realpath(__file__))
|
||||||
|
|
||||||
async def test_admin(env):
|
async def test_admin(env):
|
||||||
async with meshctrl.session.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session,\
|
async with meshctrl.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session,\
|
||||||
meshctrl.session.Session(env.mcurl, user="privileged", password=env.users["privileged"], ignore_ssl=True) as privileged_session:
|
meshctrl.Session(env.mcurl, user="privileged", password=env.users["privileged"], ignore_ssl=True) as privileged_session:
|
||||||
admin_users = await admin_session.list_users(timeout=10)
|
admin_users = await admin_session.list_users(timeout=10)
|
||||||
print("\ninfo list_users: {}\n".format(admin_users))
|
print("\ninfo list_users: {}\n".format(admin_users))
|
||||||
try:
|
try:
|
||||||
@@ -34,22 +34,22 @@ async def test_admin(env):
|
|||||||
|
|
||||||
async def test_users(env):
|
async def test_users(env):
|
||||||
try:
|
try:
|
||||||
async with meshctrl.session.Session(env.mcurl[3:], user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session:
|
async with meshctrl.Session(env.mcurl[3:], user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session:
|
||||||
pass
|
pass
|
||||||
except* ValueError:
|
except* ValueError:
|
||||||
pass
|
pass
|
||||||
else:
|
else:
|
||||||
raise Exception("Connected with bad URL")
|
raise Exception("Connected with bad URL")
|
||||||
try:
|
try:
|
||||||
async with meshctrl.session.Session(env.mcurl, user="admin", ignore_ssl=True) as admin_session:
|
async with meshctrl.Session(env.mcurl, user="admin", ignore_ssl=True) as admin_session:
|
||||||
pass
|
pass
|
||||||
except* meshctrl.exceptions.MeshCtrlError:
|
except* meshctrl.exceptions.MeshCtrlError:
|
||||||
pass
|
pass
|
||||||
else:
|
else:
|
||||||
raise Exception("Connected with no password")
|
raise Exception("Connected with no password")
|
||||||
async with meshctrl.session.Session(env.mcurl+"/", user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session,\
|
async with meshctrl.Session(env.mcurl+"/", user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session,\
|
||||||
meshctrl.session.Session(env.mcurl, user="privileged", password=env.users["privileged"], ignore_ssl=True) as privileged_session,\
|
meshctrl.Session(env.mcurl, user="privileged", password=env.users["privileged"], ignore_ssl=True) as privileged_session,\
|
||||||
meshctrl.session.Session(env.mcurl, user="unprivileged", password=env.users["unprivileged"], ignore_ssl=True) as unprivileged_session:
|
meshctrl.Session(env.mcurl, user="unprivileged", password=env.users["unprivileged"], ignore_ssl=True) as unprivileged_session:
|
||||||
|
|
||||||
assert len(await admin_session.list_users(timeout=10)) == 3, "Wrong number of users"
|
assert len(await admin_session.list_users(timeout=10)) == 3, "Wrong number of users"
|
||||||
|
|
||||||
@@ -74,17 +74,17 @@ async def test_users(env):
|
|||||||
assert len(await admin_session.list_users(timeout=10)) == 3, "Failed to remove user"
|
assert len(await admin_session.list_users(timeout=10)) == 3, "Failed to remove user"
|
||||||
|
|
||||||
async def test_login_token(env):
|
async def test_login_token(env):
|
||||||
async with meshctrl.session.Session(env.mcurl, user="unprivileged", password=env.users["unprivileged"], ignore_ssl=True) as s:
|
async with meshctrl.Session(env.mcurl, user="unprivileged", password=env.users["unprivileged"], ignore_ssl=True) as s:
|
||||||
token = await s.add_login_token("test", expire=1, timeout=10)
|
token = await s.add_login_token("test", expire=1, timeout=10)
|
||||||
print("\ninfo add_login_token: {}\n".format(token))
|
print("\ninfo add_login_token: {}\n".format(token))
|
||||||
|
|
||||||
async with meshctrl.session.Session(env.mcurl, user=token["tokenUser"], password=token["tokenPass"], ignore_ssl=True) as s2:
|
async with meshctrl.Session(env.mcurl, user=token["tokenUser"], password=token["tokenPass"], ignore_ssl=True) as s2:
|
||||||
assert (await s2.user_info())["_id"] == (await s.user_info())["_id"], "Login token logged into wrong account"
|
assert (await s2.user_info())["_id"] == (await s.user_info())["_id"], "Login token logged into wrong account"
|
||||||
# Wait for the login token to expire
|
# Wait for the login token to expire
|
||||||
await asyncio.sleep(65)
|
await asyncio.sleep(65)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
async with meshctrl.session.Session(env.mcurl, user=token["tokenUser"], password=token["tokenPass"], ignore_ssl=True) as s2:
|
async with meshctrl.Session(env.mcurl, user=token["tokenUser"], password=token["tokenPass"], ignore_ssl=True) as s2:
|
||||||
pass
|
pass
|
||||||
except:
|
except:
|
||||||
pass
|
pass
|
||||||
@@ -94,7 +94,7 @@ async def test_login_token(env):
|
|||||||
token = await s.add_login_token("test2", timeout=10)
|
token = await s.add_login_token("test2", timeout=10)
|
||||||
token2 = await s.add_login_token("test3", timeout=10)
|
token2 = await s.add_login_token("test3", timeout=10)
|
||||||
print("\ninfo add_login_token_no_expire: {}\n".format(token))
|
print("\ninfo add_login_token_no_expire: {}\n".format(token))
|
||||||
async with meshctrl.session.Session(env.mcurl, user=token["tokenUser"], password=token["tokenPass"], ignore_ssl=True) as s2:
|
async with meshctrl.Session(env.mcurl, user=token["tokenUser"], password=token["tokenPass"], ignore_ssl=True) as s2:
|
||||||
assert (await s2.user_info())["_id"] == (await s.user_info())["_id"], "Login token logged into wrong account"
|
assert (await s2.user_info())["_id"] == (await s.user_info())["_id"], "Login token logged into wrong account"
|
||||||
|
|
||||||
r = await s.list_login_tokens(timeout=10)
|
r = await s.list_login_tokens(timeout=10)
|
||||||
@@ -107,9 +107,9 @@ async def test_login_token(env):
|
|||||||
assert len(await s.remove_login_token([token2["name"]], timeout=10)) == 0, "Residual login tokens"
|
assert len(await s.remove_login_token([token2["name"]], timeout=10)) == 0, "Residual login tokens"
|
||||||
|
|
||||||
async def test_mesh_device(env):
|
async def test_mesh_device(env):
|
||||||
async with meshctrl.session.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session,\
|
async with meshctrl.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session,\
|
||||||
meshctrl.session.Session(env.mcurl, user="privileged", password=env.users["privileged"], ignore_ssl=True) as privileged_session,\
|
meshctrl.Session(env.mcurl, user="privileged", password=env.users["privileged"], ignore_ssl=True) as privileged_session,\
|
||||||
meshctrl.session.Session(env.mcurl, user="unprivileged", password=env.users["unprivileged"], ignore_ssl=True) as unprivileged_session:
|
meshctrl.Session(env.mcurl, user="unprivileged", password=env.users["unprivileged"], ignore_ssl=True) as unprivileged_session:
|
||||||
# Test creating a mesh
|
# Test creating a mesh
|
||||||
mesh = await admin_session.add_device_group("test", description="This is a test group", amtonly=False, features=0, consent=0, timeout=10)
|
mesh = await admin_session.add_device_group("test", description="This is a test group", amtonly=False, features=0, consent=0, timeout=10)
|
||||||
print("\ninfo add_device_group: {}\n".format(mesh))
|
print("\ninfo add_device_group: {}\n".format(mesh))
|
||||||
@@ -266,9 +266,9 @@ async def test_mesh_device(env):
|
|||||||
assert not (await admin_session.add_users_to_device_group((await privileged_session.user_info())["_id"], mesh.meshid, rights=meshctrl.constants.MeshRights.fullrights, timeout=5))[(await privileged_session.user_info())["_id"]]["success"], "Added user to device group which doesn't exist?"
|
assert not (await admin_session.add_users_to_device_group((await privileged_session.user_info())["_id"], mesh.meshid, rights=meshctrl.constants.MeshRights.fullrights, timeout=5))[(await privileged_session.user_info())["_id"]]["success"], "Added user to device group which doesn't exist?"
|
||||||
|
|
||||||
async def test_user_groups(env):
|
async def test_user_groups(env):
|
||||||
async with meshctrl.session.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session,\
|
async with meshctrl.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session,\
|
||||||
meshctrl.session.Session(env.mcurl, user="privileged", password=env.users["privileged"], ignore_ssl=True) as privileged_session,\
|
meshctrl.Session(env.mcurl, user="privileged", password=env.users["privileged"], ignore_ssl=True) as privileged_session,\
|
||||||
meshctrl.session.Session(env.mcurl, user="unprivileged", password=env.users["unprivileged"], ignore_ssl=True) as unprivileged_session:
|
meshctrl.Session(env.mcurl, user="unprivileged", password=env.users["unprivileged"], ignore_ssl=True) as unprivileged_session:
|
||||||
|
|
||||||
user_group = await admin_session.add_user_group("test", description="aoeu")
|
user_group = await admin_session.add_user_group("test", description="aoeu")
|
||||||
print("\ninfo add_user_group: {}\n".format(user_group))
|
print("\ninfo add_user_group: {}\n".format(user_group))
|
||||||
@@ -294,7 +294,7 @@ async def test_user_groups(env):
|
|||||||
assert await admin_session.remove_user_group(user_group2.id.split("/")[-1])
|
assert await admin_session.remove_user_group(user_group2.id.split("/")[-1])
|
||||||
|
|
||||||
async def test_events(env):
|
async def test_events(env):
|
||||||
async with meshctrl.session.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session:
|
async with meshctrl.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session:
|
||||||
await admin_session.list_events()
|
await admin_session.list_events()
|
||||||
mesh = await admin_session.add_device_group("test", description="This is a test group", amtonly=False, features=0, consent=0, timeout=10)
|
mesh = await admin_session.add_device_group("test", description="This is a test group", amtonly=False, features=0, consent=0, timeout=10)
|
||||||
try:
|
try:
|
||||||
@@ -310,7 +310,7 @@ async def test_events(env):
|
|||||||
await asyncio.sleep(1)
|
await asyncio.sleep(1)
|
||||||
else:
|
else:
|
||||||
break
|
break
|
||||||
async with meshctrl.session.Session(env.mcurl, user="privileged", password=env.users["privileged"], ignore_ssl=True) as privileged_session:
|
async with meshctrl.Session(env.mcurl, user="privileged", password=env.users["privileged"], ignore_ssl=True) as privileged_session:
|
||||||
|
|
||||||
# assert len(await privileged_session.list_events()) == 0, "non-admin user has access to admin events"
|
# assert len(await privileged_session.list_events()) == 0, "non-admin user has access to admin events"
|
||||||
|
|
||||||
@@ -337,8 +337,8 @@ async def test_events(env):
|
|||||||
assert (await admin_session.remove_device_group(mesh.meshid, timeout=10)), "Failed to remove device group"
|
assert (await admin_session.remove_device_group(mesh.meshid, timeout=10)), "Failed to remove device group"
|
||||||
|
|
||||||
async def test_interuser(env):
|
async def test_interuser(env):
|
||||||
async with meshctrl.session.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session,\
|
async with meshctrl.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session,\
|
||||||
meshctrl.session.Session(env.mcurl, user="privileged", password=env.users["privileged"], ignore_ssl=True) as privileged_session:
|
meshctrl.Session(env.mcurl, user="privileged", password=env.users["privileged"], ignore_ssl=True) as privileged_session:
|
||||||
got_message = asyncio.Event()
|
got_message = asyncio.Event()
|
||||||
async def _():
|
async def _():
|
||||||
async for message in admin_session.events({"action": "interuser"}):
|
async for message in admin_session.events({"action": "interuser"}):
|
||||||
@@ -361,7 +361,7 @@ async def test_interuser(env):
|
|||||||
tg.create_task(asyncio.wait_for(got_message.wait(), 5))
|
tg.create_task(asyncio.wait_for(got_message.wait(), 5))
|
||||||
|
|
||||||
async def test_session_files(env):
|
async def test_session_files(env):
|
||||||
async with meshctrl.session.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session:
|
async with meshctrl.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session:
|
||||||
mesh = await admin_session.add_device_group("test", description="This is a test group", amtonly=False, features=0, consent=0, timeout=10)
|
mesh = await admin_session.add_device_group("test", description="This is a test group", amtonly=False, features=0, consent=0, timeout=10)
|
||||||
try:
|
try:
|
||||||
with env.create_agent(mesh.short_meshid) as agent:
|
with env.create_agent(mesh.short_meshid) as agent:
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ import meshctrl
|
|||||||
import requests
|
import requests
|
||||||
|
|
||||||
async def test_shell(env):
|
async def test_shell(env):
|
||||||
async with meshctrl.session.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session:
|
async with meshctrl.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session:
|
||||||
mesh = await admin_session.add_device_group("test", description="This is a test group", amtonly=False, features=0, consent=0, timeout=10)
|
mesh = await admin_session.add_device_group("test", description="This is a test group", amtonly=False, features=0, consent=0, timeout=10)
|
||||||
try:
|
try:
|
||||||
with env.create_agent(mesh.short_meshid) as agent:
|
with env.create_agent(mesh.short_meshid) as agent:
|
||||||
@@ -40,7 +40,7 @@ async def test_shell(env):
|
|||||||
|
|
||||||
|
|
||||||
async def test_smart_shell(env):
|
async def test_smart_shell(env):
|
||||||
async with meshctrl.session.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session:
|
async with meshctrl.Session(env.mcurl, user="admin", password=env.users["admin"], ignore_ssl=True) as admin_session:
|
||||||
mesh = await admin_session.add_device_group("test", description="This is a test group", amtonly=False, features=0, consent=0, timeout=10)
|
mesh = await admin_session.add_device_group("test", description="This is a test group", amtonly=False, features=0, consent=0, timeout=10)
|
||||||
try:
|
try:
|
||||||
with env.create_agent(mesh.short_meshid) as agent:
|
with env.create_agent(mesh.short_meshid) as agent:
|
||||||
|
|||||||
Reference in New Issue
Block a user