From a97fdb954865b16ffeb4e7e91e2990d94d641e43 Mon Sep 17 00:00:00 2001
From: Justin Crawford <Justin@stacksmash.net>
Date: Fri, 9 Apr 2021 11:37:02 -0700
Subject: [PATCH] Display restrictions in whois information

---
 sukuinote/plugins/info.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/sukuinote/plugins/info.py b/sukuinote/plugins/info.py
index 42bd6d4..ac4d6a3 100644
--- a/sukuinote/plugins/info.py
+++ b/sukuinote/plugins/info.py
@@ -55,6 +55,12 @@ async def info(client, message):
     if entity.username:
         text_ping += f'\n<b>Username:</b> @{entity.username}'
         text_unping += f'\n<b>Username:</b> @{ZWS}{entity.username}'
+    if entity.restrictions:
+        restrictions = []
+        for r in entity.restrictions:
+            restrictions.append(f"{r.reason}-{r.platform}")
+        text_ping += f'\n<b>Restrictions:</b> {", ".join(restrictions)}'
+        text_unping += f'\n<b>Restrictions:</b> {", ".join(restrictions)}'
     if entity.members_count:
         text_ping += f'\n<b>Members:</b> {entity.members_count}'
         text_unping += f'\n<b>Members:</b> {entity.members_count}'