add html formating

telegram_export/formatters/baseformatter.py

@@ -46,6 +46,9 @@
     'id', 'name', 'mime_type', 'size', 'thumbnail_id', 'type', 'local_id',
     'volume_id', 'secret', 'extra'
 ))
+Forward = namedtuple('Forward', (
+    'id', 'original_date', 'from_id', 'channel_post', 'post_author'
+))
 
 
 class BaseFormatter:
@@ -186,7 +189,7 @@ def format(self, target, file=None, *args, **kwargs):
             file = sys.stdout
         elif isinstance(file, (str, Path)):
             if os.path.isdir(file):
-                file = os.path.join(file, str(target))
+                file = os.path.join(file, str(target)+ "." + self.name())
             file = open(file, 'w')
         elif not isinstance(file, TextIOWrapper):  # Is there a better way?
             raise TypeError(
@@ -427,5 +430,30 @@ def get_media(self, mid):
             return None
         return Media(*row)
 
+    def get_forward(self, forward_id):
+        cur = self.dbconn.cursor()
+        cur.execute("SELECT ID, OriginalDate, FromID, ChannelPost, PostAuthor FROM Forward "
+                    "WHERE ID = ?;", (forward_id,))
+        row = cur.fetchone()
+        if not row:
+            return None
+        row = row[:1] + (datetime.datetime.fromtimestamp(row[1]),) + row[2:]
+        return Forward(*row)
+
+    def get_human_readable_forwarded(self, forward):
+        if not forward:
+            return "Unknow forward"
+        if forward.from_id:
+            ent = self.get_user(forward.from_id)
+            if not ent:
+                return "Unknown user: %s" % forward.from_id
+        elif forward.channel_post:
+            ent = self.get_channel(forward.channel_post)
+            if not ent:
+                return "Unknown channel: %s" % forward.channel_post
+        else:
+            return "Unknown forward type"
+        return self.get_display_name(ent)
+
 # if __name__ == '__main__':
     # main()

telegram_export/formatters/htmlformatter.py

@@ -4,6 +4,9 @@
 """
 from . import BaseFormatter
 
+import utils as export_utils
+from .htmltemplates import *
+
 
 class HtmlFormatter(BaseFormatter):
     """A Formatter class to generate HTML"""
@@ -13,23 +16,55 @@ def name():
 
     def output_header(self, file, context):
         """Output the header of the page. Context should be a namedtuple"""
-        # TODO HTML
-        print(self.get_display_name(context), file=file)
+        name = self.get_display_name(context)
+
+        print(header.format(name, styles, name), file=file)
 
-    def generate_message_html(self, message):
+    def generate_message_html(self, message, context):
         """
         Return HTML for a message, showing reply message, forward headers,
         view count, post author, and media (if applicable).
         """
-        # TODO HTML
         from_name = self.get_display_name(message.from_id) or "(???)"
-        return "{}: {}".format(from_name, message.text)
 
-    def _format(self, context_id, file, *args, **kwargs):
+        if message.media_id:
+            media = self.get_media(message.media_id)
+            if media.type == 'photo':
+
+                media_path = "%s-%s.%s%s" % (media.type, media.name, media.id,
+                                             export_utils.get_extension(media.mime_type))
+                media_path = 'usermedia/%s-%s/%s' %(self.get_display_name(context), context.id, media_path)
+                body = photo_body_template.format(img_path=media_path)
+            else:
+                text = "Unsupported media type: %s. Message_id: %s, media_id: %s" % \
+                       (media.type, message.id, media.id)
+                body = normal_body_template.format(text=text)
+        else:
+            text = message.text
+            body = normal_body_template.format(text=text)
+        if message.reply_message_id:
+            body = reply_template.format(message_id=message.reply_message_id) + body
+        if message.forward_id:
+            forward = self.get_forward(message.forward_id)
+            body = forwaded_wrapper_template.format(
+                forward_from=self.get_human_readable_forwarded(forward),
+                body=body,
+                date=forward.original_date)
+        rendered = basic_message_template.format(
+            message_id=message.id,
+            from_name=from_name,
+            body=body,
+            full_date=message.date,
+            time=message.date.strftime("%H:%M"))
+
+        return rendered
+
+    def _format(self, context_id, file, *args, **kwargs,):
         """Format the given context as HTML and output to 'file'"""
         entity = self.get_entity(context_id)
-
+        # file = sys.stdout
         self.output_header(file, entity)
         for message in self.get_messages_from_context(context_id,
                                                       order='ASC'):
-            print(self.generate_message_html(message), file=file)
+            print(self.generate_message_html(message, entity), file=file)
+        print(footer, file=file)

telegram_export/formatters/htmltemplates.py

@@ -0,0 +1,150 @@
+column_width = 480
+styles = """
+body {
+    margin: 0;
+    font: 12px/18px 'Open Sans',"Lucida Grande","Lucida Sans Unicode",Arial,Helvetica,Verdana,sans-serif;
+}
+.page_body {
+    padding-top: 64px;
+    width: %spx;
+    margin: 0 auto;
+}
+.history {
+    padding: 16px 0;
+}
+.message {
+    margin: 0 -10px;
+    transition: background-color 2.0s ease;
+}
+.default {
+padding: 10px
+}
+.default .from_name {
+
+    color: #3892db;
+    font-weight: 700;
+    padding-bottom: 5px;
+
+}
+.details {
+
+    color: #70777b;
+
+}
+.pull_right {
+
+    float: right;
+
+}
+.photo {
+max-width: %spx;
+}
+.body {
+margin-left: 40px;
+}
+
+.page_header .content .text {
+
+    padding: 24px 24px 22px 24px;
+    font-size: 22px;
+    overflow: hidden;
+    text-overflow: ellipsis;
+    white-space: nowrap;
+
+}
+.bold {
+
+    color: #212121;
+    font-weight: 700;
+
+}
+.page_header .content {
+
+    width: 480px;
+    margin: 0 auto;
+    border-radius: 0 !important;
+
+}
+.page_header {
+
+    position: fixed;
+    z-index: 10;
+    background-color: #ffffff;
+    width: 100%%;
+    border-bottom: 1px solid #e3e6e8;
+
+}
+""" % (column_width, column_width - 20)
+
+header = """
+<html lang="en">
+<head>
+  <meta charset="utf-8">
+
+  <title>{}</title>
+  <meta name="description" content="The HTML5 Herald">
+  <meta name="author" content="SitePoint">
+  <style type="text/css">{}</style>
+
+</head>
+
+<body>
+  <div class="page_header">
+
+   <div class="content">
+
+    <div class="text bold">
+ {}
+    </div>
+
+   </div>
+
+  </div>
+<div class="page_body chat_page">
+<div class='history'>
+"""
+
+footer = """
+</div>
+</div>
+</body>
+</html>
+"""
+
+basic_message_template = """
+<div class="message default clearfix" id="message{message_id}" >
+<a name="message{message_id}"></a>
+    <div class="body">
+        <div class="pull_right date details" title="{full_date}">
+            {time}
+        </div>
+        <div class="from_name">
+            {from_name}
+        </div>
+        {body}
+    </div>
+</div>
+"""
+
+normal_body_template = """
+<div class="text">{text}</div>
+"""
+photo_body_template = """
+<div class="img_body">
+    <a href="{img_path}"> <img class='photo' src="{img_path}"></img></a>
+</div>
+"""
+
+forwaded_wrapper_template = """
+<div class="body">
+    <div class="from_name">
+        {forward_from} <span class='details'>{date}</span>
+    </div>
+    {body}
+</div>
+"""
+reply_template = """
+<div class="reply_to details">
+    In reply to <a href="#message{message_id}">this message</a>
+</div>
+"""