Skip to content

Instantly share code, notes, and snippets.

@mainframed

mainframed/ansi_to_tk4-.py

Created July 9, 2020 21:05
Show Gist options
  • Save mainframed/140fd58cc449c486e19c770a7aaaa423 to your computer and use it in GitHub Desktop.
Save mainframed/140fd58cc449c486e19c770a7aaaa423 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
ansi_to_tk4-.py
#!/usr/bin/env python3
from datetime import datetime
import sys
import re
if len(sys.argv) < 6:
print("\nTK4- ANSI art to HLASM Tool\n\nUsage:")
usage = '''{} ansi_file tk4_user source_file_member cursor_x cursor_y > JCL_FILE\n
ansi_file: ANSI art template file
tk4_user: User for Jobcard NOTIFY
source_file_member: MEMBER name for new screen
cursor_x/cursor_y: ROW/COLUMN to put cursor/input field
JCL_FILE: file to upload to TK4-
Note: You will see stderr output for information purposes
'''.format(sys.argv[0])
print(usage)
sys.exit(-1)
error_out = "[+] ANSI File:\t {}\n[+] User:\t {}\n[+] Member:\t {}\n[+] Cursor X:\t {}\n[+] Cursor Y:\t {}"
print(error_out.format(sys.argv[1],sys.argv[2],sys.argv[3],sys.argv[4],sys.argv[5]), file=sys.stderr)
ansi_file = sys.argv[1]
user = sys.argv[2]
logofile = sys.argv[3]
cursor_x = sys.argv[4]
cursor_y = sys.argv[5]
ANSI_DICT = {
'0' : 'REMOVE ATTRIBUTES',
'1' : '2902C0F842F7', # SFE HI,SKIP,WHITE
'30' : '2902C0F842F7', # SFE HI,SKIP,WHITE
'31' : "2902C0F842F2", # SFE HI,SKIP,RED
'32' : "2902C0F842F4", # SFE HI,SKIP,GREEN
'33' : "2902C0F842F6", # SFE HI,SKIP,YELLOW
'34' : "2902C0F842F1", # SFE HI,SKIP,BLUE
'35' : "2902C0F842F3", # SFE HI,SKIP,PINK ( ANSI magenta )
'36' : "2902C0F842FD", # SFE HI,SKIP,LTURQ ( ANSI cyan )
'37' : '2902C0F842F7', # SFE HI,SKIP,WHITE
}
start = '''//{user_job:<8} JOB (SETUP),
// 'Build Netsol',
// CLASS=A,
// MSGCLASS=X,
// MSGLEVEL=(1,1),
// COND=(0,NE),
// NOTIFY={user}
//********************************************************************
//*
//* Desc: Build new NETSOL logon screen: {logofile}
//* Date: {date}
//*
//* After submitting run the following to refresh VTAM in hercules
//* console:
//*
//* /Z NET,QUICK
//* /P SNASOL
//* /P JRP
//*
//* Then the commands to bring the services back up are:
//*
//* /S NET
//* /S SNASOL
//* /S JRP
//*
//********************************************************************
//*
//* First delete the previous version if it exists
//*
//CLEANUP EXEC PGM=IDCAMS
//SYSPRINT DD SYSOUT=*
//SYSIN DD *
DELETE SYS1.UMODMAC({logofile})
SET MAXCC=0
SET LASTCC=0
//*
//* Then we "compress" SYS1.UMODMAC to free up space
//*
//COMP EXEC COMPRESS,LIB='SYS1.UMODMAC'
//*
//* Next we copy NETSOL from SYS1.MACLIB to UMODMAC
//*
//COPY EXEC PGM=IEBCOPY
//SYSPRINT DD SYSOUT=*
//SYSUT1 DD DISP=SHR,DSN=SYS1.MACLIB
//SYSUT2 DD DISP=SHR,DSN=SYS1.UMODMAC
//SYSIN DD *
COPY INDD=SYSUT1,OUTDD=SYSUT2
SELECT MEMBER=((NETSOL,,R))
//*
//* Then we create {logofile} with our new welcome screen
//* After that we edit SYS1.UMODMAC(NETSOL) and make
//* Changes needed
//*
//UPDATE EXEC PGM=IEBUPDTE
//SYSPRINT DD SYSOUT=*
//SYSUT1 DD DISP=SHR,DSN=SYS1.UMODMAC
//SYSUT2 DD DISP=SHR,DSN=SYS1.UMODMAC
//SYSIN DD *
./ ADD NAME={logofile}
PUSH PRINT
PRINT OFF
TK4MLOG DS 0D
TK4MLOGW $WCC (RESETKBD,MDT)
$SBA (1,1)
DC C'Terminal'
$SBA (1,9)
$SF (SKIP,HI)
$SBA (1,11)
$SF (SKIP,HI)
TK4MDEV DC CL8' '
$SBA (1,20)
$SF (SKIP,HI)
$SBA (1,66)
$SF (SKIP,HI)
DC C'Date'
$SBA (1,71)
$SF (SKIP,HI)
$SBA (1,72)
$SF (SKIP,HI)
TK4MDATE DC CL8' '
$SBA (2,1)
$SF (SKIP,HI)
$SBA (2,66)
$SF (SKIP,HI)
DC C'Time'
$SBA (2,71)
$SF (SKIP,HI)
$SBA (2,72)
$SF (SKIP,HI)
TK4MTIME DC CL8' \''''
end = ''' $SBA (24,80)
$SF (SKIP,HI)
TK4MLOGL EQU *-TK4MLOG
POP PRINT
./ CHANGE NAME=NETSOL
CLI MSGINDEX,X'0C' , is this msg to be shown? 23164805
BNE NOUSS , bif not 23164810
* , update logo screen 23164815
LA R15,DATETIME , call DATETIME .. 23164816
LA R3,TK4MDATE , .. to fill date and .. 23164817
LA R4,TK4MTIME , .. time fields .. 23164818
BALR R14,R15 , .. on the logo screen 23164819
MVC SYNARG(FOUR),CID , get CID into SYNCHRPL 23164820
OI MFLAGS2,INQUIRE , indicate INQUIRE 23164825
OI MACFLAGS,INQCIDX , indicate INQ CIDXLATE 23164830
NI SRPLEXT1,FF-RPLNIB , synch RPL has CID in 23164835
* , ARG field 23164840
INQUIRE RPL=SYNCHRPL, , get terminal name *23164845
OPTCD=CIDXLATE, , from CID and put *23164846
AREA=TK4MDEV, , it on the logo screen *23164847
AREALEN=D8 23164848
NI MFLAGS2,FF-INQUIRE , INQUIRE is done 23164850
* , Now write the screen 23164852
LA R3,TK4MLOGL , load length of screen data 23164855
L R4,=A(TK4MLOG) , load address of screen data 23164860
WRITE RPL=(PTRRPL), , send data X23164865
OPTCD=(LBT,ERASE), , erase screen first X23164866
AREA=(R4), , address is in R4 X23164867
RECLEN=(R3), , length is in R3 X23164868
EXIT=WRITEND 23164869
B USSOK , continue normal processing 23164870
NOUSS DS 0H , issue netsol message <> 12 23164875
USSOK DS 0H , logon screen has been sent 23166010
COPY {logofile:<8} , logon screen copy book 66810010
DATETIME DS 0H adapted from Jim Morrison's U370DATE 67320010
STM R14,R12,12(R13) save registers 67320020
LR R12,R15 establish .. 67320030
USING DATETIME,R12 .. addressability 67320040
GETMAIN RU,LV=LWORKA get savearea storage 67320050
ST R13,4(,R1) chain the .. 67320060
ST R1,8(,R13) .. saveareas 67320070
LR R13,R1 establish own savearea 67320080
MVC 72(LWORKA-72,R13),WORKAREA+72 copy workarea .. 67320090
USING WORKAREA,R13 .. and establish addressability 67320100
LR R9,R3 remember address of date field on logo 67320110
TIME DEC get system date and time packed decimal 67320120
*---------------------------------------------------------------------- 67320130
* Convert HHMMSSth, YYYY to EBCDIC 67320140
*---------------------------------------------------------------------- 67320150
STM R0,R1,SAARG store packed decimal date & time 67320160
AP SAARG+4(4),P19 Y2K: add S/370 epoch century 67320170
UNPK SACHR,SAARG packed to EBCDIC 67320180
OI SACHRX,X'F0' repair sign 67320190
*---------------------------------------------------------------------- 67320200
* Convert year to binary 67320210
*---------------------------------------------------------------------- 67320220
L R3,SAARG+4 Y2K: YYYYDDDF 67320230
SRL R3,16-4 000YYYY. 67320240
ST R3,SAPAKY 67320250
OI SAPAKY+3,X'0F' packed year 67320260
CVB R3,SADWD 67320270
ST R3,SABINY binary year 67320280
*---------------------------------------------------------------------- 67320290
* Select month table 67320300
*---------------------------------------------------------------------- 67320310
LA R8,NOTLEAP not a leap year 67320320
TM SABINY+3,X'03' divisible by 4? 67320330
BC 5,CALCMON no, can't be leap year 67320340
SLR R6,R6 67320350
LA R10,400 divisible by 400 is leap year 67320360
LR R7,R3 67320370
DR R6,R10 67320380
LTR R6,R6 67320390
BZ SETLEAP evenly divisible 67320400
XR R6,R6 67320405
LA R10,100 divisible by 100 not leap year 67320410
LR R7,R3 67320420
DR R6,R10 67320430
LTR R6,R6 67320440
BZ CALCMON evenly divisible 67320450
SETLEAP LA R8,LEAP leap year 67320460
*---------------------------------------------------------------------- 67320470
* Find month & month day, given Julian days DDD in year 67320480
*---------------------------------------------------------------------- 67320490
CALCMON DS 0H R8 @ month table 67320500
LH R0,SAPAKDDD DDDF 67320510
STH R0,SAPAKD 67320520
CVB R5,SADWD2 67320530
ST R5,SABIND binary ddd 67320540
* 67320550
LA R1,1 67320560
SLR R14,R14 month minus one 67320570
SLR R15,R15 67320580
SCANMON IC R15,0(R14,R8) # days in month 67320590
CR R5,R15 too many? 67320600
BNH SETMON no, br; now know month 67320610
SR R5,R15 reduce ddd 67320620
AR R14,R1 bump month 67320630
B SCANMON 67320640
SETMON DS 0H 67320650
LA R1,100 decimal shift factor 67320660
SLR R6,R6 67320670
LA R7,1(,R14) month 67320680
MR R6,R1 67320690
AR R7,R5 binary month, day of month 67320700
CVD R7,SADWD3 decimal: 0000 0000 000M MDDF 67320710
OI SAPAKMDX,X'0F' assure reasonable sign 67320720
UNPK SACHRMD,SAPAKMD MMDD to EBCDIC 67320730
*---------------------------------------------------------------------- 67320740
* Return data to caller 67320750
*---------------------------------------------------------------------- 67320760
MVC 0(2,R9),SARESULT+14 DD R9 holds address of logo date 67320770
MVI 2(R9),C'.' 67320780
MVC 3(2,R9),SARESULT+12 MM 67320790
MVI 5(R9),C'.' 67320800
MVC 6(2,R9),SARESULT+10 YY 67320810
MVC 0(2,R4),SARESULT+0 HH R4 holds address of logo time 67320820
MVI 2(R4),C':' 67320830
MVC 3(2,R4),SARESULT+2 MM 67320840
MVI 5(R4),C':' 67320850
MVC 6(2,R4),SARESULT+4 SS 67320860
LR R1,R13 get own savearea address 67320870
L R13,4(,R13) restore caller's savearea address 67320880
FREEMAIN RU,A=(R1),LV=LWORKA free savearea storage 67320890
LM R14,R12,12(R13) restore caller's regs 67320900
SLR R15,R15 set return code of zero 67320910
BR R14 return to caller 67320920
* J F M A M J J A S O N D 67320930
NOTLEAP DC AL1(31,28,31,30,31,30,31,31,30,31,30,31) 67320940
LEAP DC AL1(31,29,31,30,31,30,31,31,30,31,30,31) 67320950
P19 DC P'1900000' packed EPOCH 67320960
WORKAREA DS 0F working storage for DATETIME 67320970
DS 18F DATETIME's save area 67320980
SADWD DS D year 67320990
SABINY EQU SADWD+0,4 binary 67321000
SAPAKY EQU SADWD+4,4 packed 000Y,YYYF 67321010
* 67321020
SADWD2 DS D julian day of year 67321030
SABIND EQU SADWD2+0,4 binary 67321040
SAPAKD EQU SADWD2+6,2 packed DDDF 67321050
* 67321060
SADWD3 DS D gregorian month, day of month 67321070
SABINMD EQU SADWD3+0,4 binary 0000MMDD 67321080
SAPAKMD EQU SADWD3+5,3 packed 0MMDDF 67321090
SAPAKMDX EQU *-1,1 sign repair 67321100
* 67321110
SAARG DS D HHMMSSth,YYYYDDDF 67321120
SAPAKDDD EQU SAARG+6,2 +0 1 2 3 4 5 6 7 67321130
* 67321140
SARESULT DS 0CL16 nearly final result 67321150
SACHR DS 0CL15 67321160
SACHRTM DS C'HHMMSSth' 67321170
SACHRY DS C'19YY' 67321180
SACHRD DS C'DDD' 67321190
SACHRX EQU *-1,1 sign repair 67321200
DS C' ' 67321210
SACHRMD EQU SACHRD,4 C'MMDD' 67321220
DS 0D align 67321230
LWORKA EQU *-WORKAREA 67321240
//*
//* With that done its time to assemble our new screen
//* We assemble SYS1.UMODMAC(NETSOL) with IFOX00
//*
//ASM EXEC PGM=IFOX00,REGION=1024K
//SYSLIB DD DISP=SHR,DSN=SYS1.UMODMAC,DCB=LRECL=32720
// DD DISP=SHR,DSN=SYS2.MACLIB
// DD DISP=SHR,DSN=SYS1.MACLIB
// DD DISP=SHR,DSN=SYS1.AMODGEN
//SYSUT1 DD UNIT=VIO,SPACE=(1700,(600,100))
//SYSUT2 DD UNIT=VIO,SPACE=(1700,(300,50))
//SYSUT3 DD UNIT=VIO,SPACE=(1700,(300,50))
//SYSPRINT DD SYSOUT=*,DCB=BLKSIZE=1089
//SYSPUNCH DD DISP=(NEW,PASS,DELETE),
// UNIT=VIO,SPACE=(TRK,(2,2)),
// DCB=(BLKSIZE=80,LRECL=80,RECFM=F)
//SYSIN DD *
ISTNSC00 CSECT ,
NETSOL SYSTEM=VS2
END ,
//*
//* Then we link it and put it in SYS1.VTAMLIB(ISTNSC00)
//*
//LKED EXEC PGM=IEWL,PARM='XREF,LIST,LET,NCAL',REGION=1024K
//SYSPRINT DD SYSOUT=*
//SYSLIN DD DISP=(OLD,DELETE,DELETE),DSN=*.ASM.SYSPUNCH
//SYSLMOD DD DISP=SHR,DSN=SYS1.VTAMLIB(ISTNSC00)
//SYSUT1 DD UNIT=VIO,SPACE=(1024,(200,20))
//*
//'''
mouse_input = '''* Insert Cursor and unprotected field
$SBA ({input_x},{input_y})
$SF (UNPROT,HI)
$IC
TK4MINP DC CL80' \''''
def print_hlasm(process_string, current_color, s_x=0, s_y=0):
print("[+] X: {:<4} Y: {:<4} Color: {:<4} String: {}".format(s_x, s_y,current_color,process_string), file=sys.stderr)
print("*")
sba = ' $SBA ({x},{y})'
DC = " DC C'{string}'"
SFE = " DC X'{sfe}'"
if len(process_string) > 52:
mid = int(abs(len(process_string)/2))
print(sba.format(x=s_x,y=s_y))
print(SFE.format(sfe=ANSI_DICT[current_color]))
print(DC.format(string=process_string.replace("'","''")[:mid-1]))
s_y = s_y + mid
process_string = process_string[mid:]
mid = int(abs(len(process_string)/2))
print(sba.format(x=s_x,y=s_y))
print(SFE.format(sfe=ANSI_DICT[current_color]))
print(DC.format(string=process_string.replace("'","''")))
escaped = False
text = False
escape = ''
x = 1
y = 1
s_x=0
s_y=0
string = ''
process_string = ''
f = open(ansi_file, "r", encoding='latin-1')
cics_map = f.read()
#Remove ANSI SAUCE record
if 'SAUCE' in cics_map[-129:]:
cics_map = cics_map[:-129]
current_color = '37'
print(start.format(date=datetime.today().strftime('%d-%m-%Y'),user_job=user+"L",user=user,logofile=logofile))
print("[+] Parsing ANSI Art",file=sys.stderr)
for byte in cics_map:
# Do stuff with byte.
if escaped:
if byte != "m" and byte != "[":
escape = escape + byte
elif byte == "m":
#escape = escape + str(byte)
escaped = False
for ansi_escape in escape.split(';'):
if ansi_escape not in ['0', '1']:
current_color = ansi_escape
elif ansi_escape == '1':
current_color = '37'
escape = ''
continue
else:
if ord(byte) == ord('\x1b'):
if text:
text = False
process_string = string
print_hlasm(process_string, current_color, s_x, s_y)
string = ''
escaped = True
continue
elif byte == "\n":
x = x + 1
y = 0
text = False
process_string = string
string = ''
continue
elif byte == "\r":
continue
elif byte != " " and not text:
s_x = x
s_y = y
text = True
string = string + byte
elif byte != " " and text:
string = string + byte
elif text and byte == " ":
text = False
process_string = string
print_hlasm(process_string, current_color, s_x, s_y)
string = ''
y += 1
if y > 80:
y = 1
x += 1
#print("LOOPED X: {} Y:{} Color: {} Char: {} ({})".format(x, y,current_color,byte, ord(byte)), file=sys.stderr)
print(mouse_input.format(input_x=cursor_x,input_y=cursor_y))
print(end.format(logofile=logofile))
print("[+] Done",file=sys.stderr)
print("[+] To complete install upload JCL_FILE to TK4- and submit it then run the following in hercules:",file=sys.stderr)
submit = '''
/Z NET,QUICK
/P SNASOL
/P JRP
Then the commands to bring the services back up are:
/S NET
/S SNASOL
/S JRP'''
print(submit,file=sys.stderr)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment