145 lines
No EOL
3.5 KiB
Python
145 lines
No EOL
3.5 KiB
Python
import logging
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
import sys
|
|
import datetime
|
|
try:
|
|
from sqlalchemy import create_engine, Column, Integer, String, DateTime
|
|
from sqlalchemy.orm import sessionmaker
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
except:
|
|
sys.exit("No SQLAlchemy.")
|
|
|
|
|
|
import json
|
|
|
|
import yaml
|
|
|
|
"""
|
|
This class describe a single Record (Rec() class) and the
|
|
records manager (RecDB() class)
|
|
"""
|
|
|
|
Base = declarative_base()
|
|
|
|
""" ************
|
|
RECORD ABSTRACTION
|
|
************ """
|
|
QUEUE = 0
|
|
RUN = 2
|
|
DONE = 4
|
|
|
|
class Rec(Base):
|
|
|
|
__tablename__ = 'rec'
|
|
id = Column(Integer, primary_key=True)
|
|
name = Column(String)
|
|
# TODO: timestamp
|
|
endtime = Column(DateTime)
|
|
starttime = Column(DateTime)
|
|
state = Column(Integer)
|
|
|
|
def __init__(self, name="", starttime=None, endtime=None, asjson=""):
|
|
self.error = 0
|
|
self.values = {"name":"","starttime":"", "endtime":""}
|
|
|
|
if len(asjson) == 0:
|
|
self.name = name
|
|
self.starttime = starttime
|
|
self.endtime = endtime
|
|
else:
|
|
#try:
|
|
# dec = json.loads( unicode(asjson) )
|
|
# dec = yaml.load( asjson )
|
|
dec = json.dumps( asjson )
|
|
# except:
|
|
# self.error = 0
|
|
print("dec %s %s" % (dec,type(dec)))
|
|
print("asjson %s %s" % (asjson,type(asjson)))
|
|
|
|
self.name = asjson[0]['name']
|
|
self.starttime = asjson[0]['starttime']
|
|
selfendtime = asjson[0]['endtime']
|
|
|
|
self.state = QUEUE
|
|
logging.info("New REC: ", self.values)
|
|
|
|
# launch the job for processing files
|
|
def start(self):
|
|
# calcola file da splittare
|
|
pass
|
|
|
|
def getvalue(self,val=None):
|
|
if val != None:
|
|
if self.values.has_key(val):
|
|
return self.values[val]
|
|
return self.values
|
|
|
|
def err(self):
|
|
return self.error
|
|
|
|
def set_run(self):
|
|
self.state = RUN
|
|
|
|
def set_done(self):
|
|
self.state = DONE
|
|
|
|
def __repr__(self):
|
|
return "<Rec('%s','%s','%s', '%s', '%s')>" % (self.id, self.name, self.starttime, self.endtime, self.state)
|
|
|
|
""" ************
|
|
RecDB
|
|
************ """
|
|
|
|
class RecDB:
|
|
def __init__(self):
|
|
|
|
self.engine = create_engine('sqlite:///techrec.db', echo=True)
|
|
self.conn = self.engine.connect()
|
|
|
|
Base.metadata.create_all(self.engine) # create Database
|
|
|
|
Session = sessionmaker(bind=self.engine)
|
|
self.session = Session()
|
|
|
|
def add(self, simplerecord):
|
|
logging.debug("New Record: %s" % simplerecord)
|
|
self.session.add( simplerecord )
|
|
self.session.commit()
|
|
|
|
# print all records
|
|
def printall(self):
|
|
print(" ---- DB ---- ")
|
|
allrecords = self.session.query(Rec).all()
|
|
|
|
print [ "R: %s" % record for record in allrecords ]
|
|
|
|
|
|
def execsql(self,sql):
|
|
records = self.conn.execute( sql )
|
|
|
|
def search(self, name="", starttime="", endtime=""):
|
|
|
|
logging.debug("Looking for %s - %s - %s" % (name, starttime, endtime) )
|
|
|
|
ret = self.session.query(Rec).filter_by(name='Mimmo1').all()
|
|
print "[SEARCH] RET (",type(ret),")" , ret
|
|
for row in ret:
|
|
print "[SEARCH] Found:", row
|
|
|
|
if __name__ == "__main__":
|
|
# text
|
|
db = RecDB()
|
|
|
|
_mytime = datetime.datetime(2014,05,24,15,16,17)
|
|
a = Rec(name="Mimmo1", starttime=_mytime, endtime=None)
|
|
|
|
db.add( a )
|
|
|
|
raw_input()
|
|
|
|
db.printall()
|
|
|
|
raw_input()
|
|
# db.search( Rec(name="Mimmo1") )
|
|
db.search( name="Mimmo1" )
|
|
|