db.py 2.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. #!/usr/bin/python3
  2. # -*- coding: utf-8 -*-
  3. # ##### BEGIN LICENSE BLOCK #####
  4. # Version: MPL 1.1/GPL 2.0/LGPL 2.1
  5. #
  6. # The contents of this file are subject to the Mozilla Public License Version
  7. # 1.1 (the "License"); you may not use this file except in compliance with
  8. # the License. You may obtain a copy of the License at
  9. # http://www.mozilla.org/MPL/
  10. #
  11. # Software distributed under the License is distributed on an "AS IS" basis,
  12. # WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  13. # for the specific language governing rights and limitations under the
  14. # License.
  15. #
  16. # The Original Code is Mozilla Universal charset detector code.
  17. #
  18. # The Initial Developer of the Original Code is
  19. # Netscape Communications Corporation.
  20. # Portions created by the Initial Developer are Copyright (C) 2001
  21. # the Initial Developer. All Rights Reserved.
  22. #
  23. # Contributor(s):
  24. # Jehan <jehan@girinstud.io>
  25. #
  26. # Alternatively, the contents of this file may be used under the terms of
  27. # either the GNU General Public License Version 2 or later (the "GPL"), or
  28. # the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  29. # in which case the provisions of the GPL or the LGPL are applicable instead
  30. # of those above. If you wish to allow use of your version of this file only
  31. # under the terms of either the GPL or the LGPL, and not to allow others to
  32. # use your version of this file under the terms of the MPL, indicate your
  33. # decision by deleting the provisions above and replace them with the notice
  34. # and other provisions required by the GPL or the LGPL. If you do not delete
  35. # the provisions above, a recipient may use your version of this file under
  36. # the terms of any one of the MPL, the GPL or the LGPL.
  37. #
  38. # ##### END LICENSE BLOCK #####
  39. import importlib
  40. import sys
  41. import os
  42. def load(charset_names):
  43. '''
  44. Load a list of charsets.
  45. This function will return a dictionary of charsets from our
  46. charset database.
  47. :param charset_names: a list of supported charset names.
  48. :return: a dictionary with all the loaded charsets.
  49. :rtype: dict
  50. '''
  51. charsets = {}
  52. # Temporarily change the search path for modules.
  53. sys_path_backup = sys.path
  54. current_dir = os.path.dirname(os.path.realpath(__file__))
  55. sys.path = [current_dir + '/../charsets']
  56. for name in charset_names:
  57. try:
  58. charset = importlib.import_module(name.lower())
  59. except ImportError:
  60. print('Unknown charset "{}": '
  61. 'file "charsets/{}.py" does not exist.'.format(name, name.lower()))
  62. exit(1)
  63. charsets[charset.name] = charset
  64. # Set back the default module paths.
  65. sys.path = sys_path_backup
  66. return charsets