From 48c824287ad3e77c4d9d5b718d4db512cf14a229 Mon Sep 17 00:00:00 2001
From: Markus Zolliker <markus.zolliker@psi.ch>
Date: Fri, 1 May 2020 15:28:59 +0200
Subject: [PATCH] allow class instead of class name in proxy_class

Change-Id: Ib3cd0113ecba80b5bbef9e31e8309861f8df5e39
Reviewed-on: https://forge.frm2.tum.de/review/c/sine2020/secop/playground/+/23036
Tested-by: JenkinsCodeReview <bjoern_pedersen@frm2.tum.de>
Reviewed-by: Enrico Faulhaber <enrico.faulhaber@frm2.tum.de>
Reviewed-by: Markus Zolliker <markus.zolliker@psi.ch>
---
 secop/proxy.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/secop/proxy.py b/secop/proxy.py
index 1791267..a66c867 100644
--- a/secop/proxy.py
+++ b/secop/proxy.py
@@ -164,7 +164,11 @@ def proxy_class(remote_class, name=None):
     remote class is <import path>.<class name> of a class used on the remote node
     if name is not given, 'Proxy' + <class name> is used
     """
-    rcls = get_class(remote_class)
+    if issubclass(remote_class, Module):
+        rcls = remote_class
+        remote_class = rcls.__name__
+    else:
+        rcls = get_class(remote_class)
     if name is None:
         name = rcls.__name__