Ejemplo n.º 1
0
Archivo: mqtt.py Proyecto: TeZzo1/MQTT
def can_access(user, topic, chip):
    now = datetime.now().time()
    pom = False
    userGroups = db.session.query(User_has_group.group_id).filter_by(user_id=user).all()     #[(1,)(2,)]
    for i in userGroups:
        timecardInGroup = db.session.query(Group_has_timecard.timecard_id).filter_by(group_id=i[0]).all()
        for j in timecardInGroup:
            timecardName = db.session.query(Timecard.timecard_head).filter_by(id=j[0]).scalar()
            timecardName = "device/" + timecardName + "/ctecka"
            if(find(topic, timecardName)):
                timecardId = j[0]
                pom = True
        if (pom):
            time_from = db.session.query(Group.access_time_from).filter_by(id=i[0]).scalar()
            time_to = db.session.query(Group.access_time_to).filter_by(id=i[0]).scalar()
            # print str(time_from) +" < "+ str(now) +" < "+ str(time_to)
            if time_from is not None or time_to is not None:
                if time_from < now < time_to:
                    pom = True
                    card = Card(card_number="", chip_number=chip, time=datetime.now().strftime("%Y-%m-%d %H:%M:%S"), id_card_reader=timecardId, id_user=user, access=pom)
                    db.session.add(card)
    if(pom==False):
        timecardId = Timecard.getIdAndName()
        for i in range(len(timecardId)):
            if find(topic, timecardId[i][1]):
                card = Card(card_number="", chip_number=chip, time=datetime.now().strftime("%Y-%m-%d %H:%M:%S"), id_card_reader=timecardId[i][0], id_user=user, access=pom)
                db.session.add(card)
    db.session.commit()
    return pom
Ejemplo n.º 2
0
Archivo: mqtt.py Proyecto: TeZzo1/MQTT
def on_message(client, userdata, msg):
    print(msg.topic+": "+str(msg.payload))
    if(msg.topic.endswith("ctecka/request")):
        code = msg.payload
        # user = User.find_by_number(code)
        user_chip = User.find_by_chip(code)
        before, sep, after = msg.topic.rpartition('/')
        if (user_chip) and (can_access(user_chip, msg.topic, code)):

            client.publish(before + sep + "potvrzeni", payload=ACCESS_ALLOWED_CODE)
            print("ACCESS ALLOWED")
        else:
            print user_chip
            if user_chip is None:
                timecardId = Timecard.getIdAndName()
                for i in range(len(timecardId)):
                    if find(msg.topic, timecardId[i][1]):
                        card = Card(card_number="", chip_number=code, time=datetime.now().strftime("%Y-%m-%d %H:%M:%S"), id_card_reader=timecardId[i][0], id_user="******", access=False)
                        db.session.add(card)
                        db.session.commit()
            client.publish(before + sep + "potvrzeni", payload=ACCESS_DENIED_CODE)
            print("ACCESS DENIED")

        code = DEFAULT_CODE