roland/python_examples
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add JQLStyle class

Browse Source
This commit is contained in:
Roland Thomas Jr
2024-06-09 00:45:21 -04:00
parent bc2fad47e2
commit 50e0eb839e
1 changed files with 196 additions and 176 deletions
Download Patch File Download Diff File Expand all files Collapse all files

372
cli/jql_pygments.py
View File

@@ -1,25 +1,19 @@
import json import json
from typing import Dict, List
from pygments.lexer import RegexLexer
from pygments.token import (
Text,
Keyword,
Operator,
Punctuation,
Name,
String,
Whitespace,
Error,
)
from prompt_toolkit import PromptSession
from prompt_toolkit.lexers import PygmentsLexer
from prompt_toolkit.styles import Style
from prompt_toolkit.completion import Completer, Completion
from prompt_toolkit.validation import Validator, ValidationError
from rich.console import Console
from rich.text import Text as RichText
from jira import JIRA from jira import JIRA
from jira.exceptions import JIRAError from jira.exceptions import JIRAError
from prompt_toolkit import PromptSession
from prompt_toolkit.completion import Completer, Completion
from prompt_toolkit.lexers import PygmentsLexer
from prompt_toolkit.styles import Style
from prompt_toolkit.validation import ValidationError, Validator
from prompt_toolkit.shortcuts import confirm
from pygments.lexer import RegexLexer
from pygments.token import Error, Keyword, Name, Operator, Punctuation, String, Text, Whitespace, _TokenType
from rich.console import Console
from rich.text import Text as RichText
class JQLLexer(RegexLexer): class JQLLexer(RegexLexer):
@@ -66,159 +60,158 @@ class JQLLexer(RegexLexer):
} }
nord_style = Style.from_dict( class JQLStyles:
{ nord: Style = Style.from_dict(
"pygments.whitespace": "#FFFFFF", {
"pygments.keyword": "#81A1C1 bold", "pygments.whitespace": "#FFFFFF",
"pygments.operator": "#EBCB8B bold", "pygments.keyword": "#81A1C1 bold",
"pygments.punctuation": "#BF616A", "pygments.operator": "#EBCB8B bold",
"pygments.name.attribute": "#B48EAD", "pygments.punctuation": "#BF616A",
"pygments.name.function": "#A3BE8C", "pygments.name.attribute": "#B48EAD",
"pygments.literal.string": "#D08770", "pygments.name.function": "#A3BE8C",
"pygments.text": "#D8DEE9", "pygments.literal.string": "#D08770",
"pygments.name.other": "#D08770", "pygments.text": "#D8DEE9",
"pygments.error": "#BF616A bold", "pygments.name.other": "#D08770",
} "pygments.error": "#BF616A bold",
)
token_styles = {
Whitespace: "#FFFFFF",
Keyword: "#81A1C1 bold",
Operator: "#EBCB8B bold",
Punctuation: "#BF616A",
Name.Attribute: "#B48EAD",
Name.Function: "#A3BE8C",
String: "#D08770",
Text: "#D8DEE9",
Error: "#BF616A bold",
Name.Other: "#D08770",
}
completion_styles = {
"Keywords": "#81A1C1 bold",
"Functions": "#A3BE8C",
"Attributes": "#B48EAD",
"Operators": "#EBCB8B bold",
"Projects": "#D08770",
}
completions ={
"Keywords": [
"A",
"AND",
"ARE",
"AS",
"AT",
"BE",
"BUT",
"BY",
"FOR",
"IF",
"INTO",
"IT",
"NO",
"NOT",
"OF",
"ON",
"OR",
"S",
"SUCH",
"T",
"THAT",
"THE",
"THEIR",
"THEN",
"THERE",
"THESE",
"THEY",
"THIS",
"TO",
"WILL",
"WITH"
],
"Functions": [
"issueHistory",
"openSprints",
"watchedIssues",
"myApproval",
"myPending",
"currentLogin",
"currentUser",
"membersOf",
"lastLogin",
"now",
"startOfDay",
"endOfDay",
"startOfWeek",
"endOfWeek",
"startOfMonth",
"endOfMonth",
"startOfYear",
"endOfYear"
],
"Attributes": [
"assignee",
"affectedVersion",
"attachments",
"comment",
"component",
"created",
"creator",
"description",
"due",
"duedate",
"filter",
"fixVersion",
"issuekey",
"labels",
"lastViewed",
"priority",
"project",
"reporter",
"resolved",
"sprint",
"status",
"statusCategory",
"summary",
"text",
"timespent",
"voter",
"watcher"
],
"Operators": [
"=",
"!=",
"<",
">",
"<=",
">=",
"~",
"!~",
"IN",
"NOT IN",
"IS",
"IS NOT",
"WAS",
"WAS IN",
"WAS NOT IN",
"WAS NOT"
],
"Projects": [
"QUANTUM",
"NEBULA",
"GALACTIC",
"STELLAR",
"AETHER",
"NOVA",
"COSMIC",
"LUNAR",
"ASTRAL",
"PHOTON"
]
} }
)
token: Dict[_TokenType, str] = {
Whitespace: "#FFFFFF",
Keyword: "#81A1C1 bold",
Operator: "#EBCB8B bold",
Punctuation: "#BF616A",
Name.Attribute: "#B48EAD",
Name.Function: "#A3BE8C",
String: "#D08770",
Text: "#D8DEE9",
Error: "#BF616A bold",
Name.Other: "#D08770",
}
completion: Dict[str, str] = {
"Keywords": "#81A1C1 bold",
"Functions": "#A3BE8C",
"Attributes": "#B48EAD",
"Operators": "#EBCB8B bold",
"Projects": "#D08770",
}
completions: Dict[str, List[str]] = {
"Keywords": [
"A",
"AND",
"ARE",
"AS",
"AT",
"BE",
"BUT",
"BY",
"FOR",
"IF",
"INTO",
"IT",
"NO",
"NOT",
"OF",
"ON",
"OR",
"S",
"SUCH",
"T",
"THAT",
"THE",
"THEIR",
"THEN",
"THERE",
"THESE",
"THEY",
"THIS",
"TO",
"WILL",
"WITH",
],
"Functions": [
"issueHistory",
"openSprints",
"watchedIssues",
"myApproval",
"myPending",
"currentLogin",
"currentUser",
"membersOf",
"lastLogin",
"now",
"startOfDay",
"endOfDay",
"startOfWeek",
"endOfWeek",
"startOfMonth",
"endOfMonth",
"startOfYear",
"endOfYear",
],
"Attributes": [
"assignee",
"affectedVersion",
"attachments",
"comment",
"component",
"created",
"creator",
"description",
"due",
"duedate",
"filter",
"fixVersion",
"issuekey",
"labels",
"lastViewed",
"priority",
"project",
"reporter",
"resolved",
"sprint",
"status",
"statusCategory",
"summary",
"text",
"timespent",
"voter",
"watcher",
],
"Operators": [
"=",
"!=",
"<",
">",
"<=",
">=",
"~",
"!~",
"IN",
"NOT IN",
"IS",
"IS NOT",
"WAS",
"WAS IN",
"WAS NOT IN",
"WAS NOT",
],
"Projects": [
"QUANTUM",
"NEBULA",
"GALACTIC",
"STELLAR",
"AETHER",
"NOVA",
"COSMIC",
"LUNAR",
"ASTRAL",
"PHOTON",
],
}
class JQLPrinter: class JQLPrinter:
@@ -232,7 +225,7 @@ class JQLPrinter:
tokens = list(JQLLexer().get_tokens(text)) tokens = list(JQLLexer().get_tokens(text))
rich_text = RichText() rich_text = RichText()
for token_type, value in tokens: for token_type, value in tokens:
style = token_styles.get(token_type, "white") style = JQLStyles.token.get(token_type, "white")
rich_text.append(value, style=style) rich_text.append(value, style=style)
return rich_text return rich_text
@@ -269,8 +262,8 @@ class JQLCompleter(Completer):
start_position=-len(text), start_position=-len(text),
display=display_text, display=display_text,
display_meta=category, display_meta=category,
style=f"fg: #D8DEE9 bg: {completion_styles.get(category, 'white')}", style=f"fg: #D8DEE9 bg: {JQLStyles.completion.get(category, 'white')}",
selected_style=f"fg: {completion_styles.get(category, 'white')} bg: #D8DEE9", selected_style=f"fg: {JQLStyles.completion.get(category, 'white')} bg: #D8DEE9",
) )
@@ -304,16 +297,28 @@ class JQLPrompt:
def create_jql_prompt_session(self): def create_jql_prompt_session(self):
completer = JQLCompleter(completions) completer = JQLCompleter(completions)
return PromptSession( return PromptSession(
message=[("#B48EAD", "JQL \u276f ")],
lexer=PygmentsLexer(JQLLexer), lexer=PygmentsLexer(JQLLexer),
style=nord_style, style=JQLStyles.nord,
completer=completer, completer=completer,
validator=JQLValidator(self.jira), validator=JQLValidator(self.jira),
rprompt="[b] Back [exit] Exit", rprompt=[
("#5E81AC bold", "[b] Back "),
("#BF616A bold", "[exit] Exit"),
],
bottom_toolbar=self.get_query_count, bottom_toolbar=self.get_query_count,
validate_while_typing=False,
) )
def get_input(self): def get_input(self):
user_input = self.session.prompt("Enter JQL: ", validate_while_typing=False) user_input = self.session.prompt()
if not user_input:
do_empty_query = confirm(
[("#EBCB8B bold", "[?] "), ("#D8DEE9 bold", "Do you want to perform an empty query?")],
suffix=[("#81A1C1 bold", " (Y/n) ")],
)
if not do_empty_query:
return
if user_input.lower() == "b": if user_input.lower() == "b":
return return
if user_input.lower() == "exit": if user_input.lower() == "exit":
@@ -340,6 +345,21 @@ class JQLPrompt:
self.console.print("Goodbye!", style="#BF616A bold") self.console.print("Goodbye!", style="#BF616A bold")
"""
[+] for additions
[-] for deletions
[~] for changes
[<] for incoming
[>] for outgoing
[✔] for success
[✖] for failure
[⚠] for warnings
[?] for questions
[!] for errors
[ℹ] for information
"""
def main(): def main():
config = load_config() config = load_config()
console = Console(color_system="truecolor") console = Console(color_system="truecolor")