Merge pull request andiwand#3 from keachi/master

PEP8 and Python library stuff

Author: andiwand

.pre-commit-config.yaml

@@ -0,0 +1,10 @@
+---
+
+repos:
+  - repo: local
+    hooks:
+      - id: isort
+        name: isort
+        entry: isort
+        language: system
+        types: [python]

leo.py

@@ -0,0 +1,132 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+import argparse
+import difflib
+import sys
+import time
+
+import requests
+from bs4 import BeautifulSoup
+
+COUNTRY_CODES = {
+    'en',
+    'de',
+    'fr',
+    'es',
+}
+LANGUAGES = {
+    'en': 'English',
+    'de': 'German',
+    'fr': 'French',
+    'es': 'Spanish',
+}
+
+URL = "http://dict.leo.org/dictQuery/m-vocab/{0}/query.xml" \
+      "?tolerMode=nof&lp={0}&lang=de&rmWords=off&rmSearch=on" \
+      "&search={1}&searchLoc=0&resultOrder=basic" \
+      "&multiwordShowSingle=on"
+URL_PRONOUNCE = 'http://dict.leo.org/media/audio/{0}.mp3'
+
+
+def similarity(s1, s2):
+    return difflib.SequenceMatcher(a=s1.lower(), b=s2.lower()).ratio()
+
+
+def fetch(phrase, lang):
+    url = URL.format(lang, phrase)
+    r = requests.get(url)
+    soup = BeautifulSoup(r.text, 'xml')
+    return soup.find_all('entry')
+
+
+def pronounce(phrase, lang):
+    entries = fetch(phrase, lang)
+    if not entries:
+        return None
+    result = []
+    for entry in entries:
+        for side in entry.find_all('side', lang='en'):
+            word = side.find('word').get_text()
+            if not side.find('pron'):
+                continue
+            url = side.find('pron')['url']
+            result.append((word, url))
+    if not result:
+        return False
+    result = [x for x in result if phrase in x[0]]
+    result.sort(key=lambda x: similarity(x[0], phrase), reverse=True)
+    pid = result[0][1]
+    url = URL_PRONOUNCE.format(pid)
+    play(url)
+    return True
+
+
+def translate(phrase, lang):
+    entries = fetch(phrase, lang)
+    if not entries:
+        return None
+
+    return [tuple([side.find('word').get_text()
+                  for side in entry.find_all('side')])
+            for entry in entries]
+
+
+def play(url):
+    try:
+        import vlc
+        player = vlc.MediaPlayer(url)
+        player.play()
+        while player.get_state() not in (
+                vlc.State.Stopped,
+                vlc.State.Ended,
+                vlc.State.Error,
+        ):
+            time.sleep(0.1)
+        player.stop()
+    except ImportError:
+        sys.stderr.write("Python library vlc not found")
+
+
+def main(argv=None):
+    parser = argparse.ArgumentParser(
+        description='leo dict cli',
+    )
+    parser.add_argument(
+        '-l',
+        '--language',
+        help='destination language (default is "%(default)s")',
+        default='en',
+    )
+    parser.add_argument(
+        '-p',
+        '--pronounce',
+        help='pronounce phrase',
+        action='store_true',
+    )
+    parser.add_argument('phrase', help='phrase to translate')
+    args = parser.parse_args(argv)
+
+    phrase, lang = args.phrase, args.language + 'de'
+
+    if args.pronounce:
+        result = pronounce(phrase, lang)
+        return 0 if result else 1
+
+    translations = translate(phrase, lang)
+    if not translations:
+        print('no translation found')
+        return 1
+
+    for trans in translations:
+        print('%s\t\t%s' % trans)
+
+    return 0
+
+
+if __name__ == '__main__':
+    rcode = main()
+    sys.exit(rcode)
+
+
+# vim: set ft=python sw=4 ts=4 et wrap tw=76:

leo/__init__.py

@@ -1,31 +1,49 @@
-#!/usr/bin/env python3
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
 
+import argparse
+import random
 import sys
 import time
-import random
-import argparse
+
 import leo
 
+
 def main(argv=None):
-    parser = argparse.ArgumentParser(description='pronunciation training')
-    parser.add_argument('-s', '--sequential', help='sequential order', action='store_true')
-    parser.add_argument('-p', '--pronounce', help='pronounce word', action='store_true')
+    parser = argparse.ArgumentParser(
+        description='pronunciation training',
+    )
+    parser.add_argument(
+        '-s',
+        '--sequential',
+        help='sequential order',
+        action='store_true',
+    )
+    parser.add_argument(
+        '-p',
+        '--pronounce',
+        help='pronounce word',
+        action='store_true',
+    )
     parser.add_argument('file', help='word list')
     args = parser.parse_args(argv)
-    
+
     lang = 'ende'
-    
+
     with open(args.file, 'r') as f:
         words = [line.strip() for line in f.readlines() if line.strip()]
-    if not args.sequential: random.shuffle(words)
-    
+    if not args.sequential:
+        random.shuffle(words)
+
     for word in words:
-        print(word, end='')
+        sys.stdout.write(word)
         input()
-        
+
         subs = [sub.strip() for sub in word.split('/')]
         for sub in subs:
-            if len(subs) > 1: print("%s:" % sub)
+            subs_len = len(subs)
+            if subs_len > 1:
+                print("%s:" % sub)
             translations = leo.translate(sub, lang)
             if not translations:
                 print("\tnot found")
@@ -37,10 +55,13 @@ def main(argv=None):
             for sub in subs[1:]:
                 time.sleep(0.3)
                 leo.pronounce(sub, lang)
-    
+
     return 0
 
+
 if __name__ == '__main__':
     exit = main()
     sys.exit(exit)
 
+
+# vim: set ft=python sw=4 ts=4 et wrap tw=76:

leo/leo.py

@@ -1,3 +1,4 @@
 beautifulsoup4>=4.3.0
 requests>=1.2.3
-python-vlc>=1.1.2
+lxml>=4.2.4
+pre-commit>=1.10.5

train.py leo/train.py

@@ -0,0 +1,35 @@
+from setuptools import find_packages, setup
+
+setup(
+    name='leo',
+    version='1.1',
+    description='dict.leo.org',
+    long_description='Leo Dictionary',
+    classifiers=[
+        'Development Status :: 4 - Beta',
+        'License :: OSI Approved :: Apache Software License',
+        'Operating System :: POSIX :: Linux',
+        'Programming Language :: Python :: 2.7',
+        'Programming Language :: Python :: 3.4',
+        'Programming Language :: Python :: 3.5',
+        'Programming Language :: Python :: 3.6'
+    ],
+    keywords='leo',
+    author='Tobias Rueetschi',
+    author_email='[email protected]',
+    url='https://github.com/keachi/pyleo',
+    license='GPLv3',
+    packages=find_packages(),
+    install_requires=[
+        'beautifulsoup4>=4.3.0',
+        'requests>=1.2.3',
+        'lxml>=4.1.1',
+    ],
+    extras_require={
+    },
+    entry_points="""
+    [console_scripts]
+    leo=leo.leo:main
+    train=leo.train:main
+    """
+)