Merge pull request #8 from eitozx/development

Added Roleplay class

animu/__init__.py

@@ -16,4 +16,5 @@
 __version__ = "0.1.0"
 
 
-from animu.client import Client, AsyncClient
+from animu.client import Client, AsyncClient
+from animu.roleplay import Roleplay

animu/client.py

@@ -2,6 +2,7 @@
 import requests
 
 from animu import model
+from .roleplay import Roleplay
 
 class Client:
     def __init__(self, token: str):
@@ -25,7 +26,7 @@ def quote(self):
         data = self._endpoint('quote')
         return model.Quote(data)
 
-    def roleplay(self, query: str):
+    def roleplay(self, query: Roleplay):
         data = self._endpoint(query)
         return model.Roleplay(data)
 
@@ -57,7 +58,7 @@ async def quote(self):
         data = await self._endpoint('quote')
         return model.Quote(data)
 
-    async def roleplay(self, query: str):
+    async def roleplay(self, query: Roleplay):
         data = await self._endpoint(query)
         return model.Roleplay(data)
 

animu/roleplay.py

@@ -0,0 +1,65 @@
+class Roleplay:
+    angry = 'angry'
+    baka = 'baka'
+    bite = 'bite'
+    blush = 'blush'
+    bonk = 'bonk'
+    bored = 'bored'
+    bully = 'bully'
+    bye = 'bye'
+    chase = 'chase'
+    cheer = 'cheer'
+    cringe = 'cringe'
+    cry = 'cry'
+    cuddle = 'cuddle'
+    dab = 'dab'
+    dance = 'dance'
+    die = 'die'
+    disgust = 'disgust'
+    facepalm = 'facepalm'
+    feed = 'feed'
+    glomp = 'glomp'
+    happy = 'happy'
+    hi = 'hi'
+    highfive = 'highfive'
+    hold = 'hold'
+    hug = 'hug'
+    kick = 'kick'
+    kill = 'kill'
+    kiss = 'kiss'
+    laugh = 'laugh'
+    lick = 'lick'
+    love = 'love'
+    lurk = 'lurk'
+    midfing = 'midfing'
+    nervous = 'nervous'
+    nom = 'nom'
+    nope = 'nope'
+    nuzzle = 'nuzzle'
+    panic = 'panic'
+    pat = 'pat'
+    peck = 'peck'
+    poke = 'poke'
+    pout = 'pout'
+    punch = 'punch'
+    run = 'run'
+    sad = 'sad'
+    shoot = 'shoot'
+    shrug = 'shrug'
+    sip = 'sip'
+    slap = 'slap'
+    sleepy = 'sleepy'
+    smile = 'smile'
+    smug = 'smug'
+    stab = 'stab'
+    stare = 'stare'
+    suicide = 'suicide'
+    tease = 'tease'
+    think = 'think'
+    thumbsup = 'thumbsup'
+    tickle = 'tickle'
+    triggered = 'triggered'
+    wag = 'wag'
+    wave = 'wave'
+    wink = 'wink'
+    yes = 'yes'

examples/roleplay.py

@@ -1,16 +1,6 @@
-token = 'ANIMU API TOKEN'
-
 import animu
-import asyncio
-
-async def test():
-
-    client = animu.Client(token)
-    object = await client.roleplay("angry")
-    """
-    Here 'angry' is the endpoint, to get the list of roleplay endpoitns
-    Please give a look at the API docs
-    """
-    print(object)
 
-asyncio.get_event_loop().run_until_complete(test())
+TOKEN = "ANIMU API TOKEN HERE"
+client = animu.Client(TOKEN)
+roleplay = client.roleplay(animu.Roleplay.angry)
+print(roleplay)