From f8a2449d7faef8be682709fd6cf86984e2afe5c6 Mon Sep 17 00:00:00 2001
From: Kovid Goyal <kovid@kovidgoyal.net>
Date: Wed, 25 Jan 2017 17:28:45 +0530
Subject: [PATCH] pep8 and code simplification

---
 src/calibre/ebooks/docx/cleanup.py |  1 -
 src/calibre/ebooks/docx/toc.py     | 14 +++-----------
 2 files changed, 3 insertions(+), 12 deletions(-)

diff --git a/src/calibre/ebooks/docx/cleanup.py b/src/calibre/ebooks/docx/cleanup.py
index 04bf96e441..3111a6cd93 100644
--- a/src/calibre/ebooks/docx/cleanup.py
+++ b/src/calibre/ebooks/docx/cleanup.py
@@ -220,4 +220,3 @@ def cleanup_markup(log, root, styles, dest_dir, detect_cover, XPath):
                     log.debug('Detected an image that looks like a cover')
                     img.getparent().remove(img)
                     return path
-
diff --git a/src/calibre/ebooks/docx/toc.py b/src/calibre/ebooks/docx/toc.py
index 1b05f2c91f..ad262904f0 100644
--- a/src/calibre/ebooks/docx/toc.py
+++ b/src/calibre/ebooks/docx/toc.py
@@ -7,6 +7,7 @@ __license__ = 'GPL v3'
 __copyright__ = '2013, Kovid Goyal <kovid at kovidgoyal.net>'
 
 from collections import namedtuple
+from itertools import count
 
 from lxml.etree import tostring
 
@@ -14,14 +15,6 @@ from calibre.ebooks.metadata.toc import TOC
 from calibre.ebooks.oeb.polish.toc import elem_to_toc_text
 
 
-class Count(object):
-
-    __slots__ = ('val',)
-
-    def __init__(self):
-        self.val = 0
-
-
 def from_headings(body, log, namespace):
     ' Create a TOC from headings in the document '
     XPath, descendants = namespace.XPath, namespace.descendants
@@ -33,13 +26,12 @@ def from_headings(body, log, namespace):
     level_item_map = {i+1:frozenset(xp(body)) for i, xp in enumerate(xpaths)}
     item_level_map = {e:i for i, elems in level_item_map.iteritems() for e in elems}
 
-    idcount = Count()
+    idcount = count()
 
     def ensure_id(elem):
         ans = elem.get('id', None)
         if not ans:
-            idcount.val += 1
-            ans = 'toc_id_%d' % idcount.val
+            ans = 'toc_id_%d' % (next(idcount) + 1)
             elem.set('id', ans)
         return ans