Downloading

Author: mvasilkov

.gitignore

@@ -1 +1,2 @@
 __pycache__/
+.ropeproject/

mystem/capslock.py

@@ -1,5 +1,6 @@
 from pathlib import Path
 import platform
+import re
 
 PACKAGES_ROOT = 'http://download.cdn.yandex.net/mystem/'
 
@@ -16,7 +17,11 @@
     ('Windows', 32): 'mystem-3.0-win7-32bit.zip',
 }
 
-MSDOS = platform.system() == 'Windows'
+RE_ARCHIVE = re.compile('(\.tar\.gz|\.zip)$')
+
+OS = platform.system()
+
+MSDOS = OS == 'Windows'
 
 OUR_ROOT = Path(__file__).parent.resolve()
 

mystem/install.py

@@ -1,7 +1,15 @@
+from pathlib import Path
 import os
+from shutil import copyfileobj
 from subprocess import run, PIPE
+import sys
+import tarfile
+from tempfile import TemporaryDirectory
+from urllib.request import urlopen
+from zipfile import ZipFile
 
-from .capslock import MYSTEM, MYSTEM_VERSION_STRING
+from .capslock import (MYSTEM, MYSTEM_DIR, MYSTEM_VERSION_STRING, OS,
+                       PACKAGES_ROOT, PACKAGES, RE_ARCHIVE)
 
 
 def is_installed():
@@ -17,3 +25,27 @@ def is_usable():
             pass
 
     return False
+
+
+def install_impl(bits=64):
+    p = (OS, bits)
+    if p not in PACKAGES:
+        raise NotImplementedError()
+
+    filetype = RE_ARCHIVE.search(PACKAGES[p]).group()
+    winrar = ZipFile if filetype == '.zip' else tarfile.open
+
+    with TemporaryDirectory() as dirname:
+        filename = Path(dirname) / ('a' + filetype)
+        url = PACKAGES_ROOT + PACKAGES[p]
+
+        print('Downloading %s' % url, file=sys.stderr)
+        print('Saving as %s' % filename, file=sys.stderr)
+
+        with urlopen(url) as a, open(filename, 'wb') as b:
+            copyfileobj(a, b)
+
+        print('Extracting to %s' % MYSTEM_DIR, file=sys.stderr)
+
+        with winrar(str(filename)) as a:
+            a.extractall(str(MYSTEM_DIR))