# HG changeset patch
# User Yuya Nishihara <yuya@tcha.org>
# Date 2018-08-12 02:43:24
# Node ID 2e578eced95816f820df50f24e4e6cf97696cbdd
# Parent  da130c5cef9014e99dc9a1a85d3578365ad64d86

patchbomb: really use BytesGenerator API

It wraps around Generator type.

diff --git a/hgext/patchbomb.py b/hgext/patchbomb.py
--- a/hgext/patchbomb.py
+++ b/hgext/patchbomb.py
@@ -142,7 +142,7 @@ configitem('patchbomb', 'to',
 if pycompat.ispy3:
     _bytesgenerator = emailgen.BytesGenerator
 else:
-    _bytesgenerator = lambda f: f
+    _bytesgenerator = emailgen.Generator
 
 # Note for extension authors: ONLY specify testedwith = 'ships-with-hg-core' for
 # extensions which SHIP WITH MERCURIAL. Non-mainline extensions should
@@ -806,8 +806,7 @@ def email(ui, repo, *revs, **opts):
         if opts.get('test'):
             ui.status(_('displaying '), subj, ' ...\n')
             ui.pager('email')
-            generator = emailgen.Generator(_bytesgenerator(ui),
-                                           mangle_from_=False)
+            generator = _bytesgenerator(ui, mangle_from_=False)
             try:
                 generator.flatten(m, 0)
                 ui.write('\n')
@@ -823,8 +822,7 @@ def email(ui, repo, *revs, **opts):
                 # Exim does not remove the Bcc field
                 del m['Bcc']
             fp = stringio()
-            generator = emailgen.Generator(_bytesgenerator(fp),
-                                           mangle_from_=False)
+            generator = _bytesgenerator(fp, mangle_from_=False)
             generator.flatten(m, 0)
             alldests = to + bcc + cc
             alldests = [encoding.strfromlocal(d) for d in alldests]