Keyboards models and PEP8

telegram/__init__.py

@@ -20,10 +20,9 @@ from video import Video
 from contact import Contact
 from location import Location
 from chataction import ChatAction
-# from inputfile import InputFile
 from userprofilephotos import UserProfilePhotos
-# from replykeyboardmarkup import ReplyKeyboardMarkup
-# from replykeyboardhide import ReplyKeyboardHide
+from replykeyboardmarkup import ReplyKeyboardMarkup
+from replykeyboardhide import ReplyKeyboardHide
 from forcereply import ForceReply
 from replymarkup import ReplyMarkup
 from error import TelegramError

telegram/bot.py

@@ -6,7 +6,8 @@
 import json
 import requests
 
-from telegram import (User, Message, Update, UserProfilePhotos, TelegramError, ReplyMarkup)
+from telegram import (User, Message, Update, UserProfilePhotos, TelegramError,
+                      ReplyMarkup)
 
 
 class Bot(object):

telegram/replykeyboardhide.py

@@ -0,0 +1,24 @@
+#!/usr/bin/env python
+
+
+import json
+from replymarkup import ReplyMarkup
+
+
+class ReplyKeyboardHide(ReplyMarkup):
+    def __init__(self,
+                 hide_keyboard=True,
+                 selective=None):
+        self.hide_keyboard = hide_keyboard
+        self.selective = selective
+
+    @staticmethod
+    def de_json(data):
+        return ReplyKeyboardHide(hide_keyboard=data.get('hide_keyboard', None),
+                                 selective=data.get('selective', None))
+
+    def to_json(self):
+        json_data = {'hide_keyboard': self.hide_keyboard}
+        if self.selective:
+            json_data['selective'] = self.selective
+        return json.dumps(json_data)

telegram/replykeyboardmarkup.py

@@ -0,0 +1,34 @@
+#!/usr/bin/env python
+
+
+import json
+from replymarkup import ReplyMarkup
+
+
+class ReplyKeyboardMarkup(ReplyMarkup):
+    def __init__(self,
+                 keyboard,
+                 resize_keyboard=None,
+                 one_time_keyboard=None,
+                 selective=None):
+        self.keyboard = keyboard
+        self.resize_keyboard = resize_keyboard
+        self.one_time_keyboard = one_time_keyboard
+        self.selective = selective
+
+    @staticmethod
+    def de_json(data):
+        return ReplyKeyboardMarkup(keyboard=data.get('keyboard', None),
+                                   resize_keyboard=data.get('resize_keyboard', None),
+                                   one_time_keyboard=data.get('one_time_keyboard', None),
+                                   selective=data.get('selective', None))
+
+    def to_json(self):
+        json_data = {'keyboard': self.keyboard}
+        if self.resize_keyboard:
+            json_data['resize_keyboard'] = self.resize_keyboard
+        if self.one_time_keyboard:
+            json_data['one_time_keyboard'] = self.one_time_keyboard
+        if self.selective:
+            json_data['selective'] = self.selective
+        return json.dumps(json_data)