mirror of
https://github.com/chubin/cheat.sh.git
synced 2026-06-20 13:16:44 +02:00
Igor Chubin
143 lines
4.0 KiB
Python
143 lines
4.0 KiB
Python
"""
|
|
Implementation of RosettaCode Adapter.
|
|
|
|
Exports:
|
|
|
|
Rosetta(Adapter)
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
import glob
|
|
import yaml
|
|
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
|
|
|
|
# pylint: disable=wrong-import-position,wrong-import-order
|
|
from globals import PATH_CHEAT_SHEETS, ROSETTA_PATH
|
|
from adapter import Adapter # pylint: disable=relative-import
|
|
# pylint: enable=wrong-import-position,wrong-import-order
|
|
|
|
class Rosetta(Adapter):
|
|
|
|
"""
|
|
Adapter for RosettaCode
|
|
"""
|
|
|
|
__section_name = 'rosetta'
|
|
_adapter_name = "rosetta"
|
|
_output_format = "code"
|
|
|
|
@staticmethod
|
|
def _load_rosetta_code_names():
|
|
answer = {}
|
|
for filename in glob.glob(os.path.join(PATH_CHEAT_SHEETS, '*/_info.yaml')):
|
|
text = open(filename, 'r').read()
|
|
data = yaml.load(text)
|
|
if data is None:
|
|
continue
|
|
lang = os.path.basename(os.path.dirname(filename))
|
|
if lang.startswith('_'):
|
|
lang = lang[1:]
|
|
if 'rosetta' in data:
|
|
answer[lang] = data['rosetta']
|
|
return answer
|
|
|
|
def _rosetta_get_list(self, query, task=None):
|
|
if query not in self._rosetta_code_name:
|
|
return []
|
|
|
|
lang = self._rosetta_code_name[query]
|
|
answer = []
|
|
if task:
|
|
glob_path = os.path.join(ROSETTA_PATH, 'Lang', lang, task, '*')
|
|
else:
|
|
glob_path = os.path.join(ROSETTA_PATH, 'Lang', lang, '*')
|
|
for filename in glob.glob(glob_path):
|
|
taskname = os.path.basename(filename)
|
|
answer.append(taskname)
|
|
|
|
answer = "".join("%s\n" % x for x in sorted(answer))
|
|
return answer
|
|
|
|
@staticmethod
|
|
def _parse_query(query):
|
|
if '/' in query:
|
|
task, subquery = query.split('/', 1)
|
|
else:
|
|
task, subquery = query, None
|
|
return task, subquery
|
|
|
|
def _get_task(self, lang, query):
|
|
if lang not in self._rosetta_code_name:
|
|
return ""
|
|
|
|
task, subquery = self._parse_query(query)
|
|
|
|
if task == ':list':
|
|
return self._rosetta_get_list(lang)
|
|
if subquery == ':list':
|
|
return self._rosetta_get_list(lang, task=task)
|
|
|
|
# if it is not a number or the number is too big, just ignore it
|
|
index = 1
|
|
if subquery:
|
|
try:
|
|
index = int(subquery)
|
|
except ValueError:
|
|
pass
|
|
|
|
lang_name = self._rosetta_code_name[lang]
|
|
|
|
tasks = sorted(glob.glob(os.path.join(ROSETTA_PATH, 'Lang', lang_name, task, '*')))
|
|
if not tasks:
|
|
return ""
|
|
|
|
if len(tasks) < index or index < 1:
|
|
index = 1
|
|
|
|
answer_filename = tasks[index-1]
|
|
answer = open(answer_filename, 'r').read()
|
|
|
|
return answer
|
|
|
|
def _starting_page(self, query):
|
|
number_of_pages = self._rosetta_get_list(query)
|
|
answer = (
|
|
"# %s pages available\n"
|
|
"# use /:list to list"
|
|
) % number_of_pages
|
|
return answer
|
|
|
|
def _get_page(self, topic, request_options=None):
|
|
|
|
if '/' not in topic:
|
|
return self._rosetta_get_list(topic)
|
|
|
|
lang, topic = topic.split('/', 1)
|
|
|
|
# this part should be generalized
|
|
# currently we just remove the name of the adapter from the path
|
|
if topic == self.__section_name:
|
|
return self._starting_page(topic)
|
|
|
|
if topic.startswith(self.__section_name + '/'):
|
|
topic = topic[len(self.__section_name + '/'):]
|
|
|
|
return self._get_task(lang, topic)
|
|
|
|
def _get_list(self, prefix=None):
|
|
return []
|
|
|
|
def get_list(self, prefix=None):
|
|
answer = [self.__section_name]
|
|
for i in self._rosetta_code_name:
|
|
answer.append('%s/%s/' % (i, self.__section_name))
|
|
return answer
|
|
|
|
def is_found(self, _):
|
|
return True
|
|
|
|
def __init__(self):
|
|
Adapter.__init__(self)
|
|
self._rosetta_code_name = self._load_rosetta_code_names()
|