# HG changeset patch
# User Yuya Nishihara <yuya@tcha.org>
# Date 1534041804 -32400
# Node ID 2e578eced95816f820df50f24e4e6cf97696cbdd
# Parent  da130c5cef9014e99dc9a1a85d3578365ad64d86
patchbomb: really use BytesGenerator API

It wraps around Generator type.

diff -r da130c5cef90 -r 2e578eced958 hgext/patchbomb.py
--- a/hgext/patchbomb.py	Sun Aug 12 14:03:22 2018 +0900
+++ b/hgext/patchbomb.py	Sun Aug 12 11:43:24 2018 +0900
@@ -142,7 +142,7 @@
 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 @@
         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 @@
                 # 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]