File: testout.py

package info (click to toggle)
aptly 1.6.2-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 49,928 kB
  • sloc: python: 10,398; sh: 252; makefile: 184
file content (23 lines) | stat: -rw-r--r-- 526 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import tempfile


class TestOut:
    def __init__(self):
        self.tmp_file = tempfile.NamedTemporaryFile(delete=False)
        self.read_pos = 0

    def fileno(self):
        return self.tmp_file.fileno()

    def write(self, text):
        self.tmp_file.write(text.encode())

    def get_contents(self):
        self.tmp_file.seek(self.read_pos, 0)
        return self.tmp_file.read().decode("utf-8")

    def close(self):
        self.tmp_file.close()

    def clear(self):
        self.read_pos = self.tmp_file.tell()