CubeSQL.Python2/cubesql.py at main · andreaspfeil/CubeSQL.Python2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
#
#         ___              __                        ____  ____     _ __
#        /   |  ____  ____/ /_______  ____ ______   / __ \/ __/__  (_) /
#       / /| | / __ \/ __  / ___/ _ \/ __ `/ ___/  / /_/ / /_/ _ \/ / /
#      / ___ |/ / / / /_/ / /  /  __/ /_/ (__  )  / ____/ __/  __/ / /
#     /_/  |_/_/ /_/\__,_/_/   \___/\__,_/____/  /_/   /_/  \___/_/_/
#
#  Product:     cubeSQL.Python2 - cubeSQL JSON database driver for Python2
#  Version:     Revision: 1.0.0, Build: 1
#  Date:        2021/06/03 21:58:48
#  Author:      Andreas Pfeil <patreon@familie-pfeil.com>
#
#  Description: Simple Python2 JSON client class for the cubeSQL database
#               server ported from sqlabs.com PHP client example, originally
#               written by Marco Bambini.
#
#  Usage:       import cubesql
#
#  License:     BEER license / MIT license
#
#  Copyright (C) 2021 by Andreas Pfeil
#
# -----------------------------------------------------------------------TAB=2

import hashlib
import json
import random
import base64
import socket
import binascii

def sha1( data, binary = False ):
  if binary:
    return hashlib.sha1( data ).digest()
  else:
    return hashlib.sha1( data ).hexdigest()

class CubeSQL:
  def __init__(self, host, name, password ):
    self.socket        = None
    self.socketTimeout = 12
    if self.open( host, name, password ) == False:
      raise "Connection error"

  def resetError( self ):
    self.errorCode    = 0
    self.errorMessage = ""

  def isError( self ):
    return self.errorCode != 0

  def sendRequest( self, json_request ):
    if self.socket != None:
      try:
        self.resetError()
        self.socket.sendall( json_request.encode( 'utf-8' ) )
        data = ""
        while( True ):
          buf  = self.socket.recv( 4096 ).decode( 'utf-8' )
          data = data + buf
          if( len( buf ) != 4096 ):
            break

        try:
          data = json.loads( data )
          if "errorCode" in data:
            self.errorCode = data[ "errorCode" ]
          if "errorMsg" in data:
            self.errorMessage = data[ "errorMsg" ]
          return data

        except json.decoder.JSONDecodeError:
          self.errorCode = -1
          self.errorMessage = json.decoder.JSONDecodeError.msg

      except socket.error, msg:
        self.errorCode = socket.error
        self.errorMessage = msg

    self.errorCode = -1
    self.errorMessage = "Socket error"

  def disconnect( self ):
    if self.socket != None:
      request = { "command": "DISCONNECT" }
      json_request = json.dumps( request )
      try:
        self.sendRequest( json_request )
        self.socket.close()
        self.socket = None
      except socket.error, msg:
        self.errorCode = socket.error
        self.errorMessage = msg

  def connect( self, host, port = 4430, timeOut = 12 ):
    try:
      self.disconnect()
      self.socket = socket.socket( socket.AF_INET, socket.SOCK_STREAM )
      self.socket.connect(( host, port ))
      return True
    except socket.error, msg:
      print( msg )
      self.errorCode = socket.error
      self.errorMessage = msg
      return False

  def open( self, host, username, password ):
    if self.connect( host ) == True:
      randpool = str( random.random() )[2:12]
      username = sha1( randpool + username )
      password = sha1( randpool + base64.b64encode( sha1( sha1( password, True ), True ) ) )
      request = { "command": "CONNECT", "username": username, "password": password, "randpool": randpool }
      json_request = json.dumps( request, sort_keys = True )
      self.sendRequest( json_request )

    return self.isError() == False

  def execute( self, sql ):
    request = { "command": "EXECUTE", "sql": sql }
    json_request = json.dumps( request )
    self.sendRequest( json_request )
    return self.isError() == False

  def select( self, query ):
    request = { "command": "SELECT", "sql": query }
    json_request = json.dumps( request )
    data = self.sendRequest( json_request )
    if self.isError():
      return None
    return data

# Additional methods:

  def use( self, database ):
    return self.execute( "USE DATABASE " + database + ";" )