create contrib submodule

This commit is contained in:
Casper da Costa-Luis 2019-11-30 22:37:04 +00:00
parent 3cd6fb0e47
commit 1277ca0619
No known key found for this signature in database
GPG Key ID: 986B408043AE090D
7 changed files with 17 additions and 13 deletions

View File

@ -769,7 +769,7 @@ A reusable canonical example is given below:
import contextlib
import sys
from tqdm import tqdm
from tqdm.utils import DummyTqdmFile # write to tqdm
from tqdm.contrib import DummyTqdmFile
@contextlib.contextmanager

View File

@ -7,6 +7,9 @@ include logo.png
include Makefile
include tox.ini
# Non-std submodules
recursive-include tqdm/contrib *.py
# Test suite
recursive-include tqdm/tests *.py
include requirements-dev.txt

View File

@ -949,7 +949,7 @@ A reusable canonical example is given below:
import contextlib
import sys
from tqdm import tqdm
from tqdm.utils import DummyTqdmFile # write to tqdm
from tqdm.contrib import DummyTqdmFile
@contextlib.contextmanager

View File

@ -15,7 +15,7 @@ from time import sleep
import contextlib
import sys
from tqdm import tqdm
from tqdm.utils import DummyTqdmFile
from tqdm.contrib import DummyTqdmFile
@contextlib.contextmanager

10
tqdm/contrib/__init__.py Normal file
View File

@ -0,0 +1,10 @@
from tqdm import tqdm
from tqdm.utils import ObjectWrapper
class DummyTqdmFile(ObjectWrapper):
"""Dummy file-like that will write to tqdm"""
def write(self, x, nolock=False):
# Avoid print() second call (useless \n)
if len(x.rstrip()) > 0:
tqdm.write(x, file=self._wrapped, nolock=nolock)

View File

@ -15,7 +15,7 @@ from tqdm import tqdm
from tqdm import trange
from tqdm import TqdmDeprecationWarning
from tqdm.std import Bar
from tqdm.utils import DummyTqdmFile
from tqdm.contrib import DummyTqdmFile
try:
from StringIO import StringIO

View File

@ -187,15 +187,6 @@ class ObjectWrapper(object):
self.wrapper_setattr('_wrapped', wrapped)
class DummyTqdmFile(ObjectWrapper):
"""Dummy file-like that will write to tqdm"""
def write(self, x, nolock=False):
from tqdm import tqdm
# Avoid print() second call (useless \n)
if len(x.rstrip()) > 0:
tqdm.write(x, file=self._wrapped, nolock=nolock)
class SimpleTextIOWrapper(ObjectWrapper):
"""
Change only `.write()` of the wrapped object by encoding the passed