calculate-utils-3-console-gui/libs_crutch/core/server/sid_pid_file.py

# -*- coding: utf-8 -*-

# Copyright 2012-2016 Mir Calculate. http://www.calculate-linux.org
#
#  Licensed under the Apache License, Version 2.0 (the "License");
#  you may not use this file except in compliance with the License.
#  You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.

import os
import time
from os import path
import pickle
from threading import Lock
from calculate.core.server.core_interfaces import CoreServiceInterface
from calculate.lib.utils.tools import ignore


class CoreWsdl(CoreServiceInterface):
    sid_pid_locker = Lock()

    @staticmethod
    def del_sid_pid(cls, sid):
        # delete conformity pid and sid of sid_pid file
        if not os.path.exists(cls.sids_pids):
            temp = open(cls.sids_pids, 'w')
            temp.close()
        sid_pid_t = cls.sids_pids + '_temp'

        with cls.sid_pid_locker:
            fd = open(cls.sids_pids, 'r')
            ft = open(sid_pid_t, 'w')
            while 1:
                try:
                    # read out on 1 record
                    list_sid = pickle.load(fd)
                except (IOError, KeyError, EOFError):
                    break
                if sid != list_sid[0]:
                    pickle.dump(list_sid, ft)
                else:
                    # end process pid = list_sid[1]
                    cls.serv_pid_kill(list_sid[1], sid)
                    # delete this of process file
                    while cls.glob_process_dict[list_sid[1]]['status'] == 1:
                        time.sleep(0.1)
                    cls.del_pid(cls, list_sid[1])
                    # delete process file
                    rm_fn = path.join(cls.pids, "%d.pid" % list_sid[1])
                    if path.exists(rm_fn):
                        with ignore(OSError):
                            os.unlink(rm_fn)
            fd.close()
            ft.close()

            ft = open(sid_pid_t, 'rb')
            fd = open(cls.sids_pids, 'wb')
            ft.seek(0)
            fd.write(ft.read())
            ft.close()
            fd.close()

            # delete sid file
            sid_file = os.path.join(cls.sids, "%d.sid" % sid)
            if os.path.exists(sid_file):
                os.unlink(sid_file)
            os.unlink(sid_pid_t)
        return 0

    @staticmethod
    def del_pid_from_sid_pid(cls, pid):
        # delete conformity pid and sid of sid_pid file
        if not os.path.exists(cls.sids_pids):
            temp = open(cls.sids_pids, 'w')
            temp.close()
        sid_pid_t = cls.sids_pids + '_temp'

        with cls.sid_pid_locker:
            fd = open(cls.sids_pids, 'r')
            ft = open(sid_pid_t, 'w')
            while 1:
                try:
                    # read out on 1 record
                    list_sid = pickle.load(fd)
                except (KeyError, EOFError, IOError):
                    break
                if pid != list_sid[1]:
                    pickle.dump(list_sid, ft)
            fd.close()
            ft.close()

            ft = open(sid_pid_t, 'rb')
            fd = open(cls.sids_pids, 'wb')
            ft.seek(0)
            fd.write(ft.read())
            ft.close()
            fd.close()

            # delete temp file
            os.unlink(sid_pid_t)
        return 0

    @staticmethod
    def find_sid_pid_file(cls, sid):
        results = []
        with cls.sid_pid_locker:
            if not os.path.exists(cls.sids_pids):
                temp = open(cls.sids_pids, 'w')
                temp.close()
            fd = open(cls.sids_pids, 'r')
            while 1:
                try:
                    # read out on 1 record
                    list_sid = pickle.load(fd)
                except (IOError, KeyError, EOFError):
                    break
                if sid == list_sid[0]:
                    results.append(list_sid[1])
            if not results:
                results.append(0)
            fd.close()
        return results

    @staticmethod
    def serv_pid_info(cls, sid, pid):
        f = 0
        results = []
        # Check pid presence and conformity sid

        with cls.sid_pid_locker:
            fd = open(cls.sids_pids, 'r')
            while 1:
                try:
                    # read out on 1 record
                    list_sid = pickle.load(fd)
                except (IOError, KeyError, EOFError):
                    break
                if sid == list_sid[0]:
                    if pid == list_sid[1]:
                        f = 1
            fd.close()

            # Get information about pid
            if f == 1:
                _t = cls.glob_process_dict[pid]
                # process id
                results.append(str(pid))
                # current state
                results.append(str(cls.glob_process_dict[pid]['status']))
                # start time
                results.append(str(cls.glob_process_dict[pid]['time']))
                # process (function) name
                results.append(str(cls.glob_process_dict[pid]['name']))
                # process soap method name
                results.append(str(cls.glob_process_dict[pid]['method_name']))
        return results

    @staticmethod
    def add_sid_pid(cls, sid, pid):
        """
        add conformity pid and sin in sid_pid file
        """
        with cls.sid_pid_locker:
            if not os.path.exists(cls.sids_pids):
                with open(cls.sids_pids, 'w'):
                    pass
            try:
                with open(cls.sids_pids, 'a') as fd:
                    pickle.dump([sid, pid], fd)
                return 0
            except (OSError, IOError, KeyError, EOFError):
                return 1