base class for telegram class

NoneGG · NoneGG · commit ce852de9a8cb · 2015-07-17T22:53:54.000+08:00
diff --git a/telegram/audio.py b/telegram/audio.py
@@ -2,9 +2,10 @@
 
 
 import json
+from .telegram_boject_base import Base
 
 
-class Audio(object):
+class Audio(Base):
     def __init__(self,
                  file_id,
                  duration,
@@ -30,6 +31,3 @@ def to_json(self):
         if self.file_size:
             json_data['file_size'] = self.file_size
         return json.dumps(json_data)
-
-    def __str__(self):
-        return self.to_json()
diff --git a/telegram/contact.py b/telegram/contact.py
@@ -2,9 +2,10 @@
 
 
 import json
+from .telegram_boject_base import Base
 
 
-class Contact(object):
+class Contact(Base):
     def __init__(self,
                  phone_number,
                  first_name,
@@ -29,7 +30,4 @@ def to_json(self):
             json_data['last_name'] = self.last_name
         if self.user_id:
             json_data['user_id'] = self.user_id
-        return json.dumps(json_data)
-
-    def __str__(self):
-        return self.to_json()
+        return json.dumps(json_data)
diff --git a/telegram/document.py b/telegram/document.py
@@ -2,9 +2,10 @@
 
 
 import json
+from .telegram_boject_base import Base
 
 
-class Document(object):
+class Document(Base):
     def __init__(self,
                  file_id,
                  thumb,
@@ -40,7 +41,4 @@ def to_json(self):
             json_data['mime_type'] = self.mime_type
         if self.file_size:
             json_data['file_size'] = self.file_size
-        return json.dumps(json_data)
-
-    def __str__(self):
-        return self.to_json()
+        return json.dumps(json_data)
diff --git a/telegram/forcereply.py b/telegram/forcereply.py
@@ -21,7 +21,4 @@ def to_json(self):
         json_data = {'force_reply': self.force_reply}
         if self.selective:
             json_data['selective'] = self.selective
-        return json.dumps(json_data)
-
-    def __str__(self):
-        return self.to_json()
+        return json.dumps(json_data)
diff --git a/telegram/groupchat.py b/telegram/groupchat.py
@@ -2,9 +2,10 @@
 
 
 import json
+from .telegram_boject_base import Base
 
 
-class GroupChat(object):
+class GroupChat(Base):
     def __init__(self,
                  id,
                  title):
@@ -19,7 +20,4 @@ def de_json(data):
     def to_json(self):
         json_data = {'id': self.id,
                      'title': self.title}
-        return json.dumps(json_data)
-
-    def __str__(self):
-        return self.to_json()
+        return json.dumps(json_data)
diff --git a/telegram/location.py b/telegram/location.py
@@ -2,9 +2,10 @@
 
 
 import json
+from .telegram_boject_base import Base
 
 
-class Location(object):
+class Location(Base):
     def __init__(self,
                  longitude,
                  latitude):
@@ -19,7 +20,4 @@ def de_json(data):
     def to_json(self):
         json_data = {'longitude': self.longitude,
                      'latitude': self.latitude}
-        return json.dumps(json_data)
-
-    def __str__(self):
-        return self.to_json()
+        return json.dumps(json_data)
diff --git a/telegram/message.py b/telegram/message.py
@@ -2,9 +2,10 @@
 
 
 import json
+from .telegram_boject_base import Base
 
 
-class Message(object):
+class Message(Base):
     def __init__(self,
                  message_id,
                  from_user,
@@ -202,7 +203,4 @@ def to_json(self):
             json_data['delete_chat_photo'] = self.delete_chat_photo
         if self.group_chat_created:
             json_data['group_chat_created'] = self.group_chat_created
-        return json.dumps(json_data)
-
-    def __str__(self):
-        return self.to_json()
+        return json.dumps(json_data)
diff --git a/telegram/photosize.py b/telegram/photosize.py
@@ -2,9 +2,10 @@
 
 
 import json
+from .telegram_boject_base import Base
 
 
-class PhotoSize(object):
+class PhotoSize(Base):
     def __init__(self,
                  file_id,
                  width,
@@ -28,7 +29,4 @@ def to_json(self):
                      'height': self.height}
         if self.file_size:
             json_data['file_size'] = self.file_size
-        return json.dumps(json_data)
-
-    def __str__(self):
-        return self.to_json()
+        return json.dumps(json_data)
diff --git a/telegram/replymarkup.py b/telegram/replymarkup.py
@@ -1,9 +1,7 @@
 #!/usr/bin/env python
+from .telegram_boject_base import Base
 
 
-class ReplyMarkup(object):
+class ReplyMarkup(Base):
     def to_json(self):
-        raise NotImplementedError
-
-    def __str__(self):
-        return self.to_json()
+        raise NotImplementedError
diff --git a/telegram/sticker.py b/telegram/sticker.py
@@ -2,9 +2,10 @@
 
 
 import json
+from .telegram_boject_base import Base
 
 
-class Sticker(object):
+class Sticker(Base):
     def __init__(self,
                  file_id,
                  width,
@@ -38,7 +39,4 @@ def to_json(self):
                      'thumb': self.thumb.to_json()}
         if self.file_size:
             json_data['file_size'] = self.file_size
-        return json.dumps(json_data)
-
-    def __str__(self):
-        return self.to_json()
+        return json.dumps(json_data)
diff --git a/telegram/telegram_boject_base.py b/telegram/telegram_boject_base.py
@@ -0,0 +1,26 @@
+from abc import ABCMeta, abstractmethod
+from telegram import TelegramError
+
+
+class Base(object):
+
+    """Base class for most telegram object"""
+
+    __metaclass__ = ABCMeta
+
+    def __str__(self):
+        return self.to_json()
+
+    def __getitem__(self, item):
+        try:
+            return self.__dict__[item]
+        except KeyError as e:
+            raise TelegramError(str(e))
+
+    @staticmethod
+    def de_json(data):
+        pass
+
+    @abstractmethod
+    def to_json(self):
+        pass
diff --git a/telegram/update.py b/telegram/update.py
@@ -2,9 +2,10 @@
 
 
 import json
+from .telegram_boject_base import Base
 
 
-class Update(object):
+class Update(Base):
     def __init__(self,
                  update_id,
                  message=None):
@@ -26,7 +27,4 @@ def to_json(self):
         json_data = {'update_id': self.update_id}
         if self.message:
             json_data['message'] = self.message.to_json()
-        return json.dumps(json_data)
-
-    def __str__(self):
-        return self.to_json()
+        return json.dumps(json_data)
diff --git a/telegram/user.py b/telegram/user.py
@@ -2,9 +2,10 @@
 
 
 import json
+from .telegram_boject_base import Base
 
 
-class User(object):
+class User(Base):
     def __init__(self,
                  id,
                  first_name,
@@ -37,7 +38,4 @@ def to_json(self):
             json_data['last_name'] = self.last_name
         if self.username:
             json_data['username'] = self.username
-        return json.dumps(json_data)
-
-    def __str__(self):
-        return self.to_json()
+        return json.dumps(json_data)
diff --git a/telegram/userprofilephotos.py b/telegram/userprofilephotos.py
@@ -2,9 +2,10 @@
 
 
 import json
+from .telegram_boject_base import Base
 
 
-class UserProfilePhotos(object):
+class UserProfilePhotos(Base):
     def __init__(self,
                  total_count,
                  photos):
@@ -32,7 +33,4 @@ def to_json(self):
             json_data['photos'] = []
             for photo in self.photos:
                 json_data['photos'].append([x.to_json() for x in photo])
-        return json.dumps(json_data)
-
-    def __str__(self):
-        return self.to_json()
+        return json.dumps(json_data)
diff --git a/telegram/video.py b/telegram/video.py
@@ -2,9 +2,10 @@
 
 
 import json
+from .telegram_boject_base import Base
 
 
-class Video(object):
+class Video(Base):
     def __init__(self,
                  file_id,
                  width,
@@ -52,7 +53,4 @@ def to_json(self):
             json_data['file_size'] = self.file_size
         if self.caption:
             json_data['caption'] = self.caption
-        return json.dumps(json_data)
-
-    def __str__(self):
-        return self.to_json()
+        return json.dumps(json_data)
