mirror of git://git.acid.vegas/archive.git
113 lines
3.5 KiB
Python
113 lines
3.5 KiB
Python
#!/usr/bin/env python
|
|
# IRC2ANSI - Developed by acidvegas in Python (https://git.acid.vegas/archive)
|
|
|
|
import re
|
|
|
|
def IRC2ANSI(text):
|
|
''' Convert IRC art to ANSI art. '''
|
|
|
|
_formats = {
|
|
'\x02': '\033[1m', # Bold
|
|
'\x1D': '\033[3m', # Italic
|
|
'\x1F': '\033[4m', # Underline
|
|
'\x16': '\033[7m', # Reverse
|
|
'\x0f': '\033[0m', # Reset
|
|
}
|
|
|
|
_colors = {
|
|
'00': ('37', '47'), # White
|
|
'01': ('30', '40'), # Black
|
|
'02': ('34', '44'), # Blue
|
|
'03': ('32', '42'), # Green
|
|
'04': ('91', '101'), # Red
|
|
'05': ('31', '41'), # Brown
|
|
'06': ('35', '45'), # Purple
|
|
'07': ('33', '43'), # Orange
|
|
'08': ('93', '103'), # Yellow
|
|
'09': ('92', '102'), # Light Green
|
|
'10': ('36', '46'), # Teal
|
|
'11': ('96', '106'), # Cyan
|
|
'12': ('94', '104'), # Light Blue
|
|
'13': ('95', '105'), # Magenta
|
|
'14': ('90', '100'), # Gray
|
|
'15': ('37', '47') # Light Gray
|
|
}
|
|
|
|
for i in range(16,100): # Add support for 99 colors
|
|
_colors[str(i)] = (str(i), str(i))
|
|
|
|
irc_color_regex = re.compile(r'\x03((?:\d{1,2})?(?:,\d{1,2})?)?')
|
|
|
|
def replace_irc_color(match):
|
|
''' Replace IRC color codes with ANSI color codes. '''
|
|
if (irc_code := match.group(1)):
|
|
codes = irc_code.split(',')
|
|
foreground = '0'+codes[0] if len(codes[0]) == 1 else codes[0]
|
|
background = ('0'+codes[1] if len(codes[1]) == 1 else codes[1]) if len(codes) == 2 else None
|
|
ansi_foreground = _colors.get(foreground, '37')[0]
|
|
ansi_background = _colors.get(background, '47')[1]
|
|
return f'\033[{ansi_foreground};{ansi_background}m' if background else f'\033[{ansi_foreground}m'
|
|
return '\033[0m' # Reset colors on no match
|
|
|
|
for item in _formats:
|
|
if item in text:
|
|
text = text.replace(item, _formats[item])
|
|
text = text.replace('\n', '\033[0m\n') # Make sure we end with a reset code
|
|
|
|
return irc_color_regex.sub(replace_irc_color, text) + '\033[0m'
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
'''A simple script to test the IRC2ANSI function. (can you tell m using vscode yet...)
|
|
Usage: ./irc2ansi.py <path>, where <path> is a file or directory containing IRC art.
|
|
When using a directory, the script will loop through all files in the directory and play them randomly, acting as a screensaver.'''
|
|
|
|
import os #
|
|
import sys # We import these libraries here so they are only loaded when running the script directly
|
|
import time # This way, the script can still be imported as module for other scripts, without loading these libraries
|
|
import pathlib # This is useful for testing the script directly and shows and example of how to use it in other scripts
|
|
import random #
|
|
|
|
PRINT_TITLES = True # Set to False to disable printing titles when looping dircetories
|
|
|
|
if len(sys.argv) == 2:
|
|
try:
|
|
import chardet
|
|
except ImportError:
|
|
raise SystemExit('missing chardet module (pip install chardet)')
|
|
|
|
if os.path.exists(sys.argv[1]):
|
|
option = sys.argv[1]
|
|
|
|
def play(path):
|
|
with open(path, 'rb') as art_file:
|
|
data = art_file.read()
|
|
enc = chardet.detect(data)['encoding']
|
|
lines = IRC2ANSI(data.decode(enc)).split('\n')
|
|
for line in lines:
|
|
print(line + ' ' + str(len(line)))
|
|
time.sleep(0.05)
|
|
|
|
if os.path.isdir(option):
|
|
if (results := list(pathlib.Path(option).rglob('*.[tT][xX][tT]'))):
|
|
while True:
|
|
random.shuffle(results)
|
|
for result in results:
|
|
play(result)
|
|
if PRINT_TITLES:
|
|
print(result)
|
|
time.sleep(3)
|
|
|
|
elif os.path.isfile(option):
|
|
play(option)
|
|
|
|
else:
|
|
raise SystemExit('invalid path')
|
|
|
|
else:
|
|
raise SystemExit('invalid path')
|
|
|
|
else:
|
|
raise SystemExit('missing or invalid arguments (usage: ./irc2ansi.py <path>)')
|