Merge pull request #52 from verata-veritatis/asset_module_dev

Asset module dev

Author: blacksabbather

pybit/__init__.py

@@ -1382,7 +1382,88 @@ def close_position(self, symbol):
 
         # Submit a market order against each open position for the same qty.
         return self.place_active_order_bulk(orders)
+    '''
+    Below are methods under https://bybit-exchange.github.io/docs/account_asset
+    '''
+    def create_internal_transfer(self, **kwargs):
+        """
+        Create internal transfer. For more information, see
+        https://bybit-exchange.github.io/docs/account_asset/#t-transfer
+        """
+
+        suffix="/asset/v1/private/transfer"
+        if self._verify_string(kwargs,'amount'):
+            return self._submit_request(
+                method='POST',
+                path=self.endpoint + suffix,
+                query=kwargs,
+                auth=True
+            )
+        else:
+            self.logger.error('amount must be in string format')
+
+    def create_subaccount_transfer(self, **kwargs):
+        """
+        Create internal transfer. For more information, see
+        https://bybit-exchange.github.io/docs/account_asset/#t-transfer
+        """
+
+        suffix="/asset/v1/private/sub-member/transfer"
+
+        if self._verify_string(kwargs, 'amount'):
+            return self._submit_request(
+                method='POST',
+                path=self.endpoint + suffix,
+                query=kwargs,
+                auth=True
+            )
+        else:
+            self.logger.error('amount must be in string format')
+
+    def query_transfer_list(self, **kwargs):
+        """
+        Create internal transfer. For more information, see
+        https://bybit-exchange.github.io/docs/account_asset/#t-transfer
+        """
+
+        suffix="/asset/v1/private/transfer/list"
+
+        return self._submit_request(
+            method='GET',
+            path=self.endpoint + suffix,
+            query=kwargs,
+            auth=True
+        )
 
+    def query_subaccount_list(self):
+        """
+        Create internal transfer. For more information, see
+        https://bybit-exchange.github.io/docs/account_asset/#t-transfer
+        """
+
+        suffix="/asset/v1/private/sub-member/member-ids"
+
+        return self._submit_request(
+            method='GET',
+            path=self.endpoint + suffix,
+            query={},
+            auth=True
+        )
+
+    def query_subaccount_transfer_list(self,**kwargs):
+        """
+        Create internal transfer. For more information, see
+        https://bybit-exchange.github.io/docs/account_asset/#t-transfer
+        """
+
+        suffix="/asset/v1/private/sub-member/transfer/list"
+
+        return self._submit_request(
+            method='GET',
+            path=self.endpoint + suffix,
+            query=kwargs,
+            auth=True
+        )
     '''
     Internal methods; signature and request submission.
     For more information about the request signature, see
@@ -1428,6 +1509,14 @@ def _auth(self, method, params, recv_window):
             bytes(_val, 'utf-8'), digestmod='sha256'
         ).hexdigest())
 
+    def _verify_string(self,params,key):
+        if key in params:
+            if not isinstance(params[key], str):
+                return False
+            else:
+                return True
+        return True
+
     def _submit_request(self, method=None, path=None, query=None, auth=False):
         """
         Submits the request to the API.