printer.py
5.22 KB
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
#!/bin/env python
# -*- coding: utf-8 -*-
__author__ = 'eduardo'
import logging
import requests
import json
from requests.exceptions import HTTPError
from sqlalchemy.orm import aliased
from .host import Host
from sqlalchemy import ForeignKey
from sqlalchemy.schema import Column
from sqlalchemy.types import String, Integer
from sqlalchemy import and_, insert, update
log = logging.getLogger()
class Printer(Host):
"""
Classe que identifica uma impressora
"""
__tablename__ = 'printer'
network_ip = Column(String(16), ForeignKey("host.network_ip"), nullable=False, primary_key=True)
model = Column(String)
serial = Column(String(50))
description = Column(String)
def __init__(self,
model=None,
serial=None,
description=None,
*args,
**kwargs
):
"""
:param counter: Contador da impressora
:param model: Modelo da impressora
:param serial: Número de série da impressora
"""
Host.__init__(self, *args, **kwargs)
self.model = model
self.serial = serial
self.description = description
def export_printer(self, server_url, session):
"""
Exporta todos os contadores para a impressora
"""
counter_list = session.query(
PrinterCounter
).filter(
PrinterCounter.network_ip == self.network_ip
).all()
for counter in counter_list:
result = counter.export_counter(server_url, session)
if result:
log.info("Contador %s para a impressora %s exportado com sucesso")
else:
log.error("Erro na remocao do contador %s para a impressora %s", counter.counter, self.network_ip)
return False
return True
class PrinterCounter(Printer):
"""
Classe que armazena o contador das impressoras
"""
__tablename__ = 'printer_counter'
network_ip = Column(String(16), ForeignKey("printer.network_ip"), nullable=False)
counter = Column(Integer, nullable=False, primary_key=True)
counter_time = Column(String(50), nullable=False, primary_key=True)
def __init__(self,
counter,
counter_time,
*args,
**kwargs
):
super(PrinterCounter, self).__init__(*args, **kwargs)
self.counter = counter
self.counter_time = counter_time
def update_counter(self, session):
"""
Atualiza contador da impressora
:param session: SQLAlchemy session
:return boolean: True if inserted
"""
results = session.query(self.__table__).filter(
and_(
self.__table__.c.counter == self.counter,
self.__table__.c.counter_time == self.counter_time
)
).first()
#print(results)
if results is None:
log.debug("Inserindo contador para impressora %s serial %s", self.network_ip, self.serial)
session.execute(
self.__table__.insert().values(
network_ip=self.network_ip,
counter=self.counter,
counter_time=self.counter_time
)
)
return True
session.execute(
Printer.__table__.update().values(
model=self.model,
description=self.description,
serial=self.serial
).where(
Printer.__table__.c.network_ip == self.network_ip
)
)
session.flush()
return False
def export_counter(self, server_url, session):
"""
Exporta contador da impressora para o Cocar
:param server_url: URL do servidor do Cocar
:param session: Sessão do banco de dados
:return: Verdadeiro ou falso dependendo do sucesso
"""
export_url = server_url + '/api/printer/' + self.network_ip
counter_json = {
'ip_address': self.network_ip,
'model': self.model,
'serial': self.serial,
'description': self.description,
'counter': self.counter,
'counter_time': int(float(self.counter_time))
}
# Envia a requisição HTTP
headers = {'content-type': 'application/json'}
response = requests.put(
export_url,
data=json.dumps(counter_json),
headers=headers
)
try:
# Check if request has gone wrong
response.raise_for_status()
except HTTPError:
# Something got wrong, raise error
log.error("Erro na insercao do contador para a impressora %s\n", self.network_ip, response.text)
return False
if response.status_code == 200:
log.info("Contador para a impressora %s com contador %s"
"exportado com sucesso", self.network_ip, self.counter)
# Remove o contador
session.remove(self)
session.flush()
return True
else:
log.error("Erro na remoção da impressora %s. Status code = %s", self.network_ip, response.status)
return False