Fix codeline span id

i is 1 for codelines and 0 for other lines; handle this correctly when
outputting from CustomHtmlFormatter.wrap().

1 files changed, 5 insertions, 2 deletions

diff --git a/pastepy/pastepy.py b/pastepy/pastepy.py
index f6b68ef..e57d406 100644
--- a/pastepy/pastepy.py
+++ b/pastepy/pastepy.py
@@ -26,8 +26,11 @@ class CustomHtmlFormatter(HtmlFormatter):
         yield 0, '<pre>'
         line = 1
         for i, t in source:
-            yield i, '<span id="codeline-{}">{}</span>'.format(i, t)
-            line += 1
+            if i == 1:
+                yield i, '<span id="codeline-{}">{}</span>'.format(line, t)
+                line += 1
+            else:
+                yield i, t
         yield 0, '</pre>'
 
 def init_lexers():