kota
initial commit
e60e568
raw
history blame
3.88 kB
'''
This file is part of PM4Py (More Info: https://pm4py.fit.fraunhofer.de).
PM4Py is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
PM4Py is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with PM4Py. If not, see <https://www.gnu.org/licenses/>.
'''
from typing import Optional, Dict, Any
from enum import Enum
from pm4py.util import exec_utils, pandas_utils
import pandas as pd
class Parameters(Enum):
CONNECTION_STRING = "connection_string"
PREFIX = "prefix"
def apply(conn, parameters: Optional[Dict[Any, Any]] = None) -> pd.DataFrame:
"""
Extracts an event log for the SAP Order-to-Cash process
Parameters
---------------
conn
(if provided) connection object to the database (offering cursors)
parameters
Parameters of the algorithm, including:
- Parameters.CONNECTION_STRING => ODBC connection string that is used (if no connection is provided)
- Parameters.PREFIX => prefix to add to the table names (example SAPSR3.)
Returns
---------------
dataframe
Pandas dataframe
"""
if parameters is None:
parameters = {}
import pm4py
connection_string = exec_utils.get_param_value(Parameters.CONNECTION_STRING, parameters, None)
if conn is None:
import pyodbc
conn = pyodbc.connect(connection_string)
prefix = exec_utils.get_param_value(Parameters.PREFIX, parameters, "")
curs = conn.cursor()
query = """
SELECT * FROM
(SELECT
T1.VBELN AS "case:concept:name",
DECODE(T1.TYPE, 'C', 'Create Sales Document', 'D', 'Create Delivery') AS "concept:name",
T1.TIMESTAMP AS "time:timestamp",
T1.USERNAME AS "org:resource"
FROM
(
SELECT
'C' AS TYPE,
VBAK.VBELN,
VBAK.ERDAT || ' ' || VBAK.ERZET AS TIMESTAMP,
VBAK.ERNAM AS USERNAME
FROM
"""+prefix+"""VBAK
UNION ALL
SELECT
'D' AS TYPE,
VBFA.VBELV AS VBELN,
LIKP.ERDAT || ' ' || LIKP.ERZET AS TIMESTAMP,
LIKP.ERNAM AS USERNAME
FROM
"""+prefix+"""LIKP
JOIN """+prefix+"""VBFA ON LIKP.VBELN = VBFA.VBELN
WHERE
VBFA.VBTYP_N = 'J'
) T1 )
UNION
(
SELECT
VBFA.VBELN AS "case:concept:name",
TSTCT.TTEXT AS "concept:name",
CDHDR.UDATE || ' ' || CDHDR.UTIME AS "time:timestamp",
CDHDR.USERNAME AS "org:resource"
FROM
"""+prefix+"""VBFA VBFA
JOIN """+prefix+"""VBAK VBAK ON VBFA.VBELN = VBAK.VBELN
JOIN """+prefix+"""VBAP VBAP ON VBAK.VBELN = VBAP.VBELN
JOIN """+prefix+"""CDHDR CDHDR ON VBFA.VBELN = CDHDR.OBJECTID
JOIN """+prefix+"""TSTCT TSTCT ON CDHDR.TCODE = TSTCT.TCODE
WHERE
VBFA.VBTYP_N = 'C'
AND TSTCT.SPRSL = 'E'
)
UNION
(
SELECT
C.VBELN AS "case:concept:name",
'Create Billing' AS "concept:name",
C.FKDAT AS "time:timestamp",
C.ERNAM AS "org:resource"
FROM
"""+prefix+"""VBRK C
JOIN
"""+prefix+"""VBAK A ON A.VBELN = C.VBELN
)
"""
columns = ["case:concept:name", "concept:name", "time:timestamp", "org:resource"]
curs.execute(query)
dataframe = curs.fetchall()
dataframe = pandas_utils.instantiate_dataframe_from_records(dataframe, columns=columns)
dataframe = pm4py.format_dataframe(dataframe)
curs.close()
conn.close()
return dataframe