mirror of
https://github.com/d0zingcat/RSSHub-python.git
synced 2026-06-09 23:16:57 +00:00
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
import re
|
||
import csv
|
||
import random
|
||
import requests
|
||
import linecache
|
||
from os import path
|
||
import requests
|
||
from rsshub.utils import DEFAULT_HEADERS
|
||
|
||
|
||
file_path = path.dirname(path.realpath(__file__))
|
||
|
||
def get_csv_line(url):
|
||
response = requests.get(url)
|
||
lines = response.text.splitlines()
|
||
reader = csv.reader(lines)
|
||
data = list(reader)
|
||
data = data[1:]
|
||
random_line = random.choice(data)
|
||
return random_line
|
||
|
||
def remove_html_tags(text):
|
||
"""Remove html tags from a string"""
|
||
clean = re.compile('<.*?>')
|
||
return re.sub(clean, '', text)
|
||
|
||
def ctx(category=''):
|
||
word = ''
|
||
if category == 'ja':
|
||
url = 'https://raw.githubusercontent.com/henrylovemiller/img/main/words.csv'
|
||
res = get_csv_line(url)
|
||
word = f"{res[1]} 〔{res[2]} {res[4]}〕 {res[3]} "
|
||
elif category == 'jlpt3':
|
||
url = 'https://raw.githubusercontent.com/henrylovemiller/img/main/hongbaoshu_N3.csv'
|
||
res = get_csv_line(url)
|
||
word = f"{res[0]}〔{res[1]} {res[2]}〕 ➡{res[3]} ➡{res[4]} ➡ {res[5]} ➡ {res[6]} "
|
||
word = remove_html_tags(word)
|
||
else:
|
||
file = path.join(file_path,'toeflwords.txt')
|
||
with open(file, encoding='utf-8') as inf:
|
||
f = inf.readlines()
|
||
count = len(f)
|
||
wordnum = random.randrange(0, count, 1)
|
||
word = linecache.getline(file, wordnum)
|
||
return {"word": word} |