From 1ebd6facaeed8163bf12b8329555869188e20cc3 Mon Sep 17 00:00:00 2001 From: Bel LaPointe Date: Sun, 18 Sep 2022 12:44:06 -0600 Subject: [PATCH] random reader --- src/main.py | 1 + src/reader.py | 18 ++++++++++++++++-- 2 files changed, 17 insertions(+), 2 deletions(-) diff --git a/src/main.py b/src/main.py index c4881e8..c820842 100644 --- a/src/main.py +++ b/src/main.py @@ -4,6 +4,7 @@ import bucket def main(): r = reader.Reader(reader.StdinReader()) + r = reader.Reader(reader.RandomReader()) w = writer.Writer() bkt = bucket.Bucket() for line in r.read(): diff --git a/src/reader.py b/src/reader.py index 06a27d6..7d44ced 100644 --- a/src/reader.py +++ b/src/reader.py @@ -3,6 +3,7 @@ import queue import time import sys import select +import random __interval__ = .1 @@ -36,7 +37,10 @@ class Reader: try: while not reader.stopping: try: - yield q.get(timeout=__interval__) + s = q.get(timeout=__interval__) + if s: + s = s.lower().strip() + yield s except queue.Empty: yield None pass @@ -50,11 +54,21 @@ class StdinReader: try: if select.select([sys.stdin,],[],[],__interval__/2.0)[0]: line = sys.stdin.readline() - return line.strip() + return line except Exception as e: pass return None +class RandomReader: + def read(self): + seed = random.randint(0, 99) + if seed < 10: + return "a" + elif seed < 50: + return "B" + else: + return "b" + class FileReader: def __init__(self, path): self.f = open(path, "r")