1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
# Copyright 2011 Florent Le Coz <louiz@louiz.org>
#
# This file is part of Poezio.
#
# Poezio is free software: you can redistribute it and/or modify
# it under the terms of the zlib license. See the COPYING file.
"""
Defines the Fifo class
"""
import logging
log = logging.getLogger(__name__)
import os
import threading
class OpenTrick(threading.Thread):
"""
A threaded trick to make the open for writing succeed.
A fifo cannot be opened for writing if it has not been
yet opened by the other hand for reading.
So, we just open the fifo for reading and close it
immediately afterwards.
Once that is done, we can freely keep the fifo open for
writing and write things in it. The writing can fail if
there’s still nothing reading that fifo, but we just yell
an error in that case.
"""
def __init__(self, path):
threading.Thread.__init__(self)
self.path = path
self.fd = None
def run(self):
self.fd = open(self.path, 'r', encoding='utf-8')
class Fifo(object):
"""
Just a simple file handler, writing and reading in a fifo.
Mode is either 'r' or 'w', just like the mode for the open()
function.
"""
def __init__(self, path, mode):
self.trick = None
if not os.path.exists(path):
os.mkfifo(path)
if mode == 'w':
self.trick = OpenTrick(path)
# that thread will wait until we open it for writing
self.trick.start()
self.fd = open(path, mode, encoding='utf-8')
def write(self, data):
"""
Try to write on the fifo. If that fails, this means
that nothing has that fifo opened, so the writing is useless,
so we just return (and display an error telling that, somewhere).
"""
self.fd.write(data)
self.fd.flush()
def readline(self):
return self.fd.readline()
def __del__(self):
try:
self.fd.close()
except:
pass
|