Updating the db record to allow the column selection

Author: krugazul

includes/classes/class-helpers.php

@@ -577,17 +577,18 @@ public function get_the_user_ip() {
 	 * @param string $code
 	 * @return object
 	 */
-	public function get_db_record( $code ) {
+	public function get_db_record( $code, $column = 'txn_code' ) {
 		global $wpdb;
 		$return = false;
 		$table  = $wpdb->prefix . PFF_PAYSTACK_TABLE;
 		$record = $wpdb->get_results(
 			$wpdb->prepare(
 					"SELECT * 
 					FROM %i 
-					WHERE txn_code = %s"
+					WHERE %i = %s"
 				,
 				$table,
+				$column,
 				$code
 			), 'OBJECT' );